ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-02-24 21:23:55 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix issue with A100 GPU monitoring
Some checks failed
CodeQL / Analyze (python) (push) Has been cancelled
Details

Browse Source
This commit is contained in:
clearml 2025-02-18 15:30:27 +02:00
parent c46f254839
commit 342e1b35f8
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
clearml/utilities/resource_monitor.py
View File

@ -486,7 +486,8 @@ class ResourceMonitor(BackgroundMonitor):
# only monitor the active gpu's, if none were selected, monitor everything # only monitor the active gpu's, if none were selected, monitor everything
if self._skip_nonactive_gpu(g): if self._skip_nonactive_gpu(g):
continue continue
stats["gpu_%d_temperature" % i] = float(g["temperature.gpu"]) if g["temperature.gpu"] is not None:
stats["gpu_%d_temperature" % i] = float(g["temperature.gpu"])
if g["utilization.gpu"] is not None: if g["utilization.gpu"] is not None:
stats["gpu_%d_utilization" % i] = float(g["utilization.gpu"]) stats["gpu_%d_utilization" % i] = float(g["utilization.gpu"])
else: else: