From 1bb0002b7198877efd17dbd2fcabddc364178b3b Mon Sep 17 00:00:00 2001 From: Xuehai Pan Date: Thu, 22 Sep 2022 20:00:21 +0800 Subject: [PATCH] chore(core/process): handle gone processes in `update_gpu_status` Signed-off-by: Xuehai Pan --- nvitop/core/process.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/nvitop/core/process.py b/nvitop/core/process.py index a98461d..00092f1 100644 --- a/nvitop/core/process.py +++ b/nvitop/core/process.py @@ -666,6 +666,8 @@ class GpuProcess: # pylint: disable=too-many-instance-attributes,too-many-publi def update_gpu_status(self) -> Union[int, NaType]: """Updates the GPU consumption status from a new NVML query.""" + self.set_gpu_memory(NA) + self.set_gpu_utilization(NA, NA, NA, NA) self.device.processes.cache_clear() self.device.processes() return self.gpu_memory()