ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-01-31 00:56:57 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix NVIDIA_VISIBLE_DEVICES=all env var value will result in no GPU metrics being reported

Browse Source
This commit is contained in:
allegroai 2024-02-15 19:05:25 +02:00
parent 8746f34409
commit 7362955983
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
clearml/utilities/resource_monitor.py
View File

@ -52,7 +52,7 @@ class ResourceMonitor(BackgroundMonitor):
try: try:
active_gpus = os.environ.get('NVIDIA_VISIBLE_DEVICES', '') or \ active_gpus = os.environ.get('NVIDIA_VISIBLE_DEVICES', '') or \
os.environ.get('CUDA_VISIBLE_DEVICES', '') os.environ.get('CUDA_VISIBLE_DEVICES', '')
if active_gpus: if active_gpus != "all":
self._active_gpus = [g.strip() for g in active_gpus.split(',')] self._active_gpus = [g.strip() for g in active_gpus.split(',')]
except Exception: except Exception:
pass pass