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