ClearML/clearml-agent
1
0
Fork 0
mirror of https://github.com/clearml/clearml-agent synced 2025-02-07 13:26:08 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix --gpus all not reporting GPU stats on worker machine

Browse Source
This commit is contained in:
allegroai 2022-10-07 15:22:13 +03:00
parent 1e3646b57c
commit 67fc884895
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
clearml_agent/helper/resource_monitor.py
View File

@ -92,9 +92,10 @@ class ResourceMonitor(object):
# None means no filtering, report all gpus # None means no filtering, report all gpus
self._active_gpus = None self._active_gpus = None
try: try:
active_gpus = Session.get_nvidia_visible_env() or "" active_gpus = Session.get_nvidia_visible_env()
if active_gpus: # None means no filtering, report all gpus
self._active_gpus = [g.strip() for g in active_gpus.split(',')] if active_gpus and active_gpus != "all":
self._active_gpus = [g.strip() for g in str(active_gpus).split(',')]
except Exception: except Exception:
pass pass