Skip to content

Commit 2c05858

Browse files
four4fishtchaton
andcommitted
Update pytorch_lightning/trainer/trainer.py
Co-authored-by: thomas chaton <[email protected]>
1 parent a5117e0 commit 2c05858

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pytorch_lightning/trainer/trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1638,7 +1638,7 @@ def _log_device_info(self) -> None:
16381638
num_ipus = self.ipus if self.ipus is not None else 0
16391639
rank_zero_info(f"IPU available: {_IPU_AVAILABLE}, using: {num_ipus} IPUs")
16401640

1641-
if torch.cuda.is_available() and isinstance(self.accelerator, GPUAccelerator):
1641+
if torch.cuda.is_available() and not isinstance(self.accelerator, GPUAccelerator):
16421642
rank_zero_warn(
16431643
"GPU available but not used. Set the gpus flag in your trainer `Trainer(gpus=1)` or script `--gpus=1`.",
16441644
category=PossibleUserWarning,

0 commit comments

Comments
 (0)