administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-03 17:55:21 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Get correct nvidia %

Browse Source
This commit is contained in:
Nick Mowen 2022-09-24 19:03:08 -06:00
parent 1d7265f453
commit 987bc050ed
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
frigate/util.py
View File

@ -817,7 +817,7 @@ def get_nvidia_gpu_stats() -> dict[str, str]:
"""Get stats using nvidia-smi.""" """Get stats using nvidia-smi."""
nvidia_smi_command = [ nvidia_smi_command = [
"nvidia-smi", "nvidia-smi",
"--query-gpu=gpu_name,utilization.gpu,utilization.memory", "--query-gpu=gpu_name,utilization.gpu,memory.used,memory.total",
"--format=csv" "--format=csv"
] ]
@ -832,10 +832,11 @@ def get_nvidia_gpu_stats() -> dict[str, str]:
return None return None
else: else:
usages = p.stdout.split("\n")[1].strip() 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] = { results: dict[str, str] = {
"name": usages[0], "name": usages[0],
"gpu_usage": usages[1], "gpu_usage": usages[1],
"memory_usage": usages[2] "memory_usage": memory_percent
} }
return results return results