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

Use nvidia stats

Browse Source
This commit is contained in:
Nick Mowen 2022-09-24 18:39:07 -06:00
parent 4ffdfe66ba
commit cc150147d1
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
frigate/stats.py
View File

@ -13,7 +13,7 @@ from frigate.comms.dispatcher import Dispatcher
from frigate.config import FrigateConfig from frigate.config import FrigateConfig
from frigate.const import DRIVER_AMD, DRIVER_ENV_VAR, RECORD_DIR, CLIPS_DIR, CACHE_DIR from frigate.const import DRIVER_AMD, DRIVER_ENV_VAR, RECORD_DIR, CLIPS_DIR, CACHE_DIR
from frigate.types import StatsTrackingTypes, CameraMetricsTypes from frigate.types import StatsTrackingTypes, CameraMetricsTypes
from frigate.util import get_amd_gpu_stats from frigate.util import get_amd_gpu_stats, get_nvidia_gpu_stats
from frigate.version import VERSION from frigate.version import VERSION
from frigate.util import get_cpu_stats from frigate.util import get_cpu_stats
from frigate.object_detection import ObjectDetectProcess from frigate.object_detection import ObjectDetectProcess
@ -93,9 +93,10 @@ def get_gpu_stats(config: FrigateConfig) -> dict[str, str]:
if "cuvid" in args: if "cuvid" in args:
# nvidia GPU # nvidia GPU
gpu["name"] = "nvidia-cuvid" nvidia_usage = get_nvidia_gpu_stats()
gpu["usage"] = "100"
gpu["memory"] = "200" if nvidia_usage:
stats[nvidia_usage["name"]] = nvidia_usage
elif "qsv" in args: elif "qsv" in args:
# intel QSV GPU # intel QSV GPU
gpu["name"] = "intel-qsv" gpu["name"] = "intel-qsv"