From 2685e20782fcf858e2cc1e3e0378afa21e6a5b3c Mon Sep 17 00:00:00 2001 From: zmy Date: Thu, 17 Dec 2020 20:02:34 +0800 Subject: [PATCH] bug fix --- gpuutil/gpuutil.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/gpuutil/gpuutil.py b/gpuutil/gpuutil.py index ec1c663..af6c934 100644 --- a/gpuutil/gpuutil.py +++ b/gpuutil/gpuutil.py @@ -288,10 +288,13 @@ class GPUStat(): if key != "gpu": self.simplified_info[key] = self.detailed_info[key] 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.driver_version = self.simplified_info["driver_version"] - self.attached_gpus = self.simplified_info["attached_gpus"] + self.simplified_info["gpus"] = [simplify_gpu_info(stat) for stat in self.detailed_info["gpu"]] + if "cuda_version" in self.simplified_info: + self.cuda_version = self.simplified_info["cuda_version"] + 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 = [] for i, gpu in enumerate(self.simplified_info["gpus"]): for process in gpu['processes']: