mirror of
https://github.com/blakeblackshear/frigate.git
synced 2026-02-03 17:55:21 +03:00
Add util to get radeontop results
This commit is contained in:
parent
3180dfd617
commit
497ebf63f2
@ -734,6 +734,37 @@ def escape_special_characters(path: str) -> str:
|
|||||||
# path does not have user:pass
|
# path does not have user:pass
|
||||||
return path
|
return path
|
||||||
|
|
||||||
|
def get_amd_gpu_stats() -> dict[str, str]:
|
||||||
|
"""Get stats using radeontop."""
|
||||||
|
radeontop_command = [
|
||||||
|
"radeontop",
|
||||||
|
"-d",
|
||||||
|
"-",
|
||||||
|
"-l",
|
||||||
|
"1"
|
||||||
|
]
|
||||||
|
|
||||||
|
p = sp.run(
|
||||||
|
radeontop_command,
|
||||||
|
encoding="ascii",
|
||||||
|
capture_output=True,
|
||||||
|
)
|
||||||
|
|
||||||
|
if p.returncode != 0:
|
||||||
|
logger.error(p.stderr)
|
||||||
|
return None
|
||||||
|
else:
|
||||||
|
usages = p.stdout.split(",")
|
||||||
|
results: dict[str, str] = {}
|
||||||
|
|
||||||
|
for hw in usages:
|
||||||
|
if "gpu" in hw:
|
||||||
|
results["gpu_usage"] = hw.strip().split(" ")[1]
|
||||||
|
elif "vram" in hw:
|
||||||
|
results["gpu_usage"] = hw.strip().split(" ")[1]
|
||||||
|
|
||||||
|
return results
|
||||||
|
|
||||||
|
|
||||||
def get_cpu_stats() -> dict[str, dict]:
|
def get_cpu_stats() -> dict[str, dict]:
|
||||||
"""Get cpu usages for each process id"""
|
"""Get cpu usages for each process id"""
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user