diff --git a/abogen/is_nvidia.py b/abogen/is_nvidia.py index 5ae457a..2777f7b 100644 --- a/abogen/is_nvidia.py +++ b/abogen/is_nvidia.py @@ -1,7 +1,11 @@ import gpustat def check(): - stats = gpustat.new_query() + try: + stats = gpustat.new_query() + except Exception: + return False + for gpu in stats.gpus: print(gpu.name) if 'nvidia' in gpu.name.lower(): @@ -9,4 +13,4 @@ def check(): return False if __name__ == "__main__": - print(check()) + print(check()) \ No newline at end of file