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

fix a bug

Browse Source
This commit is contained in:
mingyang 2020-12-17 19:55:29 +08:00
parent 27604247af
commit e06555d14f
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
gpuutil/gpuutil.py
View File

@ -283,12 +283,12 @@ class GPUStat():
value = [value] value = [value]
self.detailed_info[key] = [parse_gpu_info(info) for info in value] self.detailed_info[key] = [parse_gpu_info(info) for info in value]
self.process_info = self.get_process_info() self.process_info = self.get_process_info()
self.simplified_info = { self.simplified_info = {}
"driver_version": self.detailed_info["driver_version"], for key in self.detailed_info:
"cuda_version": self.detailed_info["cuda_version"], if key != "gpu":
"attached_gpus": self.detailed_info["attached_gpus"], self.simplified_info[key] = self.detailed_info[key]
"gpus": [simplify_gpu_info(stat) for stat in self.detailed_info["gpu"]] else:
} self.simplified_info[key] = [simplify_gpu_info(stat) for stat in self.detailed_info["gpu"]]
self.cuda_version = self.simplified_info["cuda_version"] self.cuda_version = self.simplified_info["cuda_version"]
self.driver_version = self.simplified_info["driver_version"] self.driver_version = self.simplified_info["driver_version"]
self.attached_gpus = self.simplified_info["attached_gpus"] self.attached_gpus = self.simplified_info["attached_gpus"]