diff --git a/frigate/util.py b/frigate/util.py index d3597be06..5d0858170 100755 --- a/frigate/util.py +++ b/frigate/util.py @@ -926,9 +926,15 @@ def get_nvidia_gpu_stats() -> dict[str, str]: "--format=csv", ] - if "CUDA_VISIBLE_DEVICES" in os.environ and os.environ["CUDA_VISIBLE_DEVICES"].isdigit(): + 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 and os.environ["NVIDIA_VISIBLE_DEVICES"].isdigit(): + 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(