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

Formatting

Browse Source
This commit is contained in:
Nick Mowen 2022-09-24 19:33:35 -06:00
parent beb8229e4d
commit 447df28746
1 changed files with 6 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
frigate/util.py
View File

@ -783,13 +783,7 @@ def ffprobe_stream(path: str) -> sp.CompletedProcess:
def get_amd_gpu_stats() -> dict[str, str]:
"""Get stats using radeontop."""
radeontop_command = [
"radeontop",
"-d",
"-",
"-l",
"1"
]
radeontop_command = ["radeontop", "-d", "-", "-l", "1"]
p = sp.run(
radeontop_command,
@ -812,18 +806,10 @@ def get_amd_gpu_stats() -> dict[str, str]:
return results
def get_intel_gpu_stats() -> dict[str, str]:
"""Get stats using intel_gpu_top."""
radeontop_command = [
"timeout",
"1s",
"intel_gpu_top",
"-J",
"-o",
"-",
"-s",
"1"
]
radeontop_command = ["timeout", "1s", "intel_gpu_top", "-J", "-o", "-", "-s", "1"]
p = sp.run(
radeontop_command,
@ -852,7 +838,7 @@ def get_nvidia_gpu_stats() -> dict[str, str]:
nvidia_smi_command = [
"nvidia-smi",
"--query-gpu=gpu_name,utilization.gpu,memory.used,memory.total",
"--format=csv"
"--format=csv",
]
p = sp.run(
@ -866,11 +852,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)}" %"
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": memory_percent
"memory_usage": memory_percent,
}
return results