Skip to content

Commit d636bd7

Browse files
four4fishtchaton
andauthored
Update pytorch_lightning/trainer/trainer.py
Co-authored-by: thomas chaton <[email protected]>
1 parent f02be7c commit d636bd7

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
@@ -1662,7 +1662,7 @@ def _log_device_info(self) -> None:
16621662
num_ipus = self.ipus if self.ipus is not None else 0
16631663
rank_zero_info(f"IPU available: {_IPU_AVAILABLE}, using: {num_ipus} IPUs")
16641664

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

0 commit comments

Comments
 (0)