ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-05-09 15:10:38 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix GPU metrics collection broken by annotations

Browse Source
This commit is contained in:
clearml 2025-04-25 16:11:46 +03:00
parent 9541a99762
commit 960165d425
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
clearml/utilities/gpu/gpustat.py
View File

@ -392,10 +392,10 @@ class GPUStatCollection(object):
return b.decode() # for python3, to unicode return b.decode() # for python3, to unicode
return b return b
def get_gpu_info(index: int, handle: N.nvmlDevice_t, is_mig: bool = False) -> dict: def get_gpu_info(index: int, handle: Any, is_mig: bool = False) -> dict:
"""Get one GPU information specified by nvml handle""" """Get one GPU information specified by nvml handle"""
def get_process_info(nv_process: N.nvmlProcessInfo) -> dict: def get_process_info(nv_process: Any) -> dict:
"""Get the process information of specific pid""" """Get the process information of specific pid"""
process = {} process = {}
if nv_process.pid not in GPUStatCollection.global_processes: if nv_process.pid not in GPUStatCollection.global_processes: