diff --git a/frigate/util.py b/frigate/util.py index 69ead2a7a..734ec6d31 100755 --- a/frigate/util.py +++ b/frigate/util.py @@ -926,6 +926,11 @@ def get_nvidia_gpu_stats() -> dict[str, str]: "--format=csv", ] + if ("CUDA_VISIBLE_DEVICES" in os.environ): + nvidia_smi_command.extend(["--id", os.environ['CUDA_VISIBLE_DEVICES']]) + elif ("NVIDIA_VISIBLE_DEVICES" in os.environ): + nvidia_smi_command.extend(["--id", os.environ['NVIDIA_VISIBLE_DEVICES']]) + p = sp.run( nvidia_smi_command, encoding="ascii",