From 987bc050ed2d75f841dd963e6882ef88205bb2c0 Mon Sep 17 00:00:00 2001 From: Nick Mowen Date: Sat, 24 Sep 2022 19:03:08 -0600 Subject: [PATCH] Get correct nvidia % --- frigate/util.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/frigate/util.py b/frigate/util.py index 32bd5bcaf..7e22a18e2 100755 --- a/frigate/util.py +++ b/frigate/util.py @@ -817,7 +817,7 @@ def get_nvidia_gpu_stats() -> dict[str, str]: """Get stats using nvidia-smi.""" nvidia_smi_command = [ "nvidia-smi", - "--query-gpu=gpu_name,utilization.gpu,utilization.memory", + "--query-gpu=gpu_name,utilization.gpu,memory.used,memory.total", "--format=csv" ] @@ -832,10 +832,11 @@ def get_nvidia_gpu_stats() -> dict[str, str]: return None else: usages = p.stdout.split("\n")[1].strip() + memory_percent = f"{round(float(usages[2].replace(' MiB', '')) / float(usages[3].replace(' MiB', '')) * 100, 1)}" %" results: dict[str, str] = { "name": usages[0], "gpu_usage": usages[1], - "memory_usage": usages[2] + "memory_usage": memory_percent } return results