take any nvidia-smi exception as not having gpu

DataBiosphere · Oct 3, 2023 · 328b3eb · 328b3eb
1 parent 9b09d62
commit 328b3eb
Showing 1 changed file with 1 addition and 7 deletions.
diff --git a/src/toil/lib/accelerators.py b/src/toil/lib/accelerators.py
@@ -100,13 +100,7 @@ def count_nvidia_gpus() -> int:
                 .firstChild,
             ).data
         )
-    except (
-        FileNotFoundError,
-        subprocess.CalledProcessError,
-        IndexError,
-        ValueError,
-        PermissionError,
-    ):
+    except: 
         return 0
 
     # TODO: Parse each gpu > product_name > text content and convert to some