bf16 not available for the current CUDA device #10716
Unanswered
ncuxomun
asked this question in
Lightning Trainer API: Trainer, LightningModule, LightningDataModule
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
Hi, guys.
I am having issues setting bf16 in my Trainer, which is set as:

The error that I get is:
RuntimeError: Current CUDA Device does not support bfloat16. Please switch dtype to float16.
At the same time, I seem to have the bf16 available:

What am I doing wrong here?
Thank you for your attention!
Beta Was this translation helpful? Give feedback.
All reactions