github-starred/nvitop
2
0
Fork 0
mirror of https://github.com/XuehaiPan/nvitop.git synced 2026-05-21 06:45:24 -06:00
Code Issues 29 Projects Packages Wiki Activity Actions

fix(callbacks): fix a potential attribute errors for newer PyTorch-Lightning

Browse source
This commit is contained in:
Xuehai Pan 2023-08-13 22:15:56 +08:00
parent d71eb5103a
commit c12b01701b
2 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

1
docs/source/spelling_wordlist.txt
View file

@ -150,3 +150,4 @@ KiB
tx tx
rx rx
ThroughputInfo ThroughputInfo
pytorch

6
nvitop/callbacks/pytorch_lightning.py
View file

@ -119,7 +119,11 @@ class GpuStatsLogger(Callback): # pylint: disable=too-many-instance-attributes
f'The root device type is {trainer.strategy.root_device.type}.', f'The root device type is {trainer.strategy.root_device.type}.',
) )
device_ids = trainer.data_parallel_device_ids try:
device_ids = trainer.device_ids # pytorch-lightning >= 1.6.0
except AttributeError:
device_ids = trainer.data_parallel_device_ids # pytorch-lightning < 1.6.0
try: try:
self._devices = get_devices_by_logical_ids(device_ids, unique=True) self._devices = get_devices_by_logical_ids(device_ids, unique=True)
except (libnvml.NVMLError, RuntimeError) as ex: except (libnvml.NVMLError, RuntimeError) as ex: