Skip to content

Instantly share code, notes, and snippets.

@SharathHebbar
Created January 24, 2024 05:50
Show Gist options
  • Save SharathHebbar/7d3e39e22680c754f12f8f2e4d5ef4fc to your computer and use it in GitHub Desktop.
Save SharathHebbar/7d3e39e22680c754f12f8f2e4d5ef4fc to your computer and use it in GitHub Desktop.

ValueError: Your setup doesn't support bf16/gpu. You need torch>=1.10, using Ampere GPU with cuda>=11.0

Change bf16 to fp16 for non Ampere GPUs

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment