Update pytorch_lightning/trainer/trainer.py

Co-authored-by: thomas chaton <thomas@grid.ai>
Lightning-AI · Dec 9, 2021 · d636bd7 · d636bd7
1 parent f02be7c
commit d636bd7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py
@@ -1662,7 +1662,7 @@ def _log_device_info(self) -> None:
         num_ipus = self.ipus if self.ipus is not None else 0
         rank_zero_info(f"IPU available: {_IPU_AVAILABLE}, using: {num_ipus} IPUs")
 
-        if torch.cuda.is_available() and isinstance(self.accelerator, GPUAccelerator):
+        if torch.cuda.is_available() and not isinstance(self.accelerator, GPUAccelerator):
             rank_zero_warn(
                 "GPU available but not used. Set the gpus flag in your trainer `Trainer(gpus=1)` or script `--gpus=1`.",
                 category=PossibleUserWarning,