diff --git a/frigate/util/services.py b/frigate/util/services.py index 1fcb31077..59cc4d27d 100644 --- a/frigate/util/services.py +++ b/frigate/util/services.py @@ -15,7 +15,12 @@ import psutil import py3nvml.py3nvml as nvml import requests -from frigate.const import DRIVER_AMD, DRIVER_ENV_VAR, FFMPEG_HWACCEL_NVIDIA, FFMPEG_HWACCEL_VAAPI +from frigate.const import ( + DRIVER_AMD, + DRIVER_ENV_VAR, + FFMPEG_HWACCEL_NVIDIA, + FFMPEG_HWACCEL_VAAPI, +) from frigate.util.builtin import clean_camera_user_pass, escape_special_characters logger = logging.getLogger(__name__) @@ -195,6 +200,7 @@ def get_bandwidth_stats(config) -> dict[str, dict]: return usages + def is_vaapi_amd_driver() -> bool: driver = os.environ.get(DRIVER_ENV_VAR) if driver: @@ -207,6 +213,7 @@ def is_vaapi_amd_driver() -> bool: return False else: output = p.stdout.decode('unicode_escape').split("\n") + # VA Info will print out the friendly name of the driver return any("AMD Radeon Graphics" in line for line in output)