zmyme/gpuutil
1
0
Fork 0
Code Issues Pull Requests Releases 5 Wiki Activity

bug fix

Browse Source
This commit is contained in:
mingyang 2020-12-17 20:02:34 +08:00
parent e06555d14f
commit 2685e20782
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
gpuutil/gpuutil.py
View File

@ -288,10 +288,13 @@ class GPUStat():
if key != "gpu": if key != "gpu":
self.simplified_info[key] = self.detailed_info[key] self.simplified_info[key] = self.detailed_info[key]
else: else:
self.simplified_info[key] = [simplify_gpu_info(stat) for stat in self.detailed_info["gpu"]] self.simplified_info["gpus"] = [simplify_gpu_info(stat) for stat in self.detailed_info["gpu"]]
self.cuda_version = self.simplified_info["cuda_version"] if "cuda_version" in self.simplified_info:
self.driver_version = self.simplified_info["driver_version"] self.cuda_version = self.simplified_info["cuda_version"]
self.attached_gpus = self.simplified_info["attached_gpus"] if "driver_version" in self.simplified_info:
self.driver_version = self.simplified_info["driver_version"]
if "attached_gpus" in self.simplified_info:
self.attached_gpus = self.simplified_info["attached_gpus"]
self.gpus = [] self.gpus = []
for i, gpu in enumerate(self.simplified_info["gpus"]): for i, gpu in enumerate(self.simplified_info["gpus"]):
for process in gpu['processes']: for process in gpu['processes']: