Fix test for BF16 detection (#17803)

src/transformers/training_args.py

@@ -1042,7 +1042,7 @@ class TrainingArguments:
             if self.no_cuda and not is_torch_bf16_cpu_available():
                 # cpu
                 raise ValueError("Your setup doesn't support bf16/cpu. You need torch>=1.10")
-            elif not is_torch_bf16_gpu_available():
+            elif not self.no_cuda and not is_torch_bf16_gpu_available():
                 # gpu
                 raise ValueError(
                     "Your setup doesn't support bf16/gpu. You need torch>=1.10, using Ampere GPU with cuda>=11.0"