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(gui/device): fix CUDA_VISIBLE_DEVICES enumeration

Browse source 
Signed-off-by: Xuehai Pan <XuehaiPan@pku.edu.cn>
This commit is contained in:
Xuehai Pan 2022-07-22 20:09:20 +08:00
parent 6533954896
commit 94be8b30d4
1 changed files with 1 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
nvitop/gui/screens/main/device.py
View file

@ -225,12 +225,11 @@ class DevicePanel(Displayable): # pylint: disable=too-many-instance-attributes
separator_line = separator_line[:-1] + '┼' + '─' * (remaining_width - 1) + '┤'
if self.device_count > 0:
for device in self.devices:
for mig_device_count in self.mig_device_counts:
if compact:
frame.extend((data_line, separator_line))
else:
frame.extend((data_line, data_line, separator_line))
mig_device_count = self.mig_device_counts[device.physical_index]
if mig_device_count > 0:
frame.extend((data_line,) * mig_device_count + (separator_line,))