handle no nvidia-smi

Author: gletort

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "fishfeats"
-version = "1.4.2"
+version = "1.4.3"
 description = "Napari plugin for RNA-Fish+cells analysis pipeline"
 license.file = "LICENSE"
 readme = "README.md"

src/fish_feats/Utils.py

@@ -803,17 +803,20 @@ def get_env_name():
 def get_cuda_feature():
     """ Get the drivers cuda version through nvidia-smi and build the relevant feature for pixi """
     import subprocess, re
-    result = subprocess.run(['nvidia-smi'], capture_output=True, text=True)
-    match = re.search(r'CUDA Version:\s+([\d.]+)', result.stdout)
-    if match:
-        cuda_version = match.group(1)
-
-        ## get only major version
-        if cuda_version.split( "." )[0] == "12":
-            return "cu126"
-        if cuda_version.split( "." )[0] == "13":
-            return "cu130"
-    return "cpu"
+    try:
+        result = subprocess.run(['nvidia-smi'], capture_output=True, text=True)
+        match = re.search(r'CUDA Version:\s+([\d.]+)', result.stdout)
+        if match:
+            cuda_version = match.group(1)
+
+            ## get only major version
+            if cuda_version.split( "." )[0] == "12":
+                return "cu126"
+            if cuda_version.split( "." )[0] == "13":
+                return "cu130"
+        return "cpu"
+    except:
+        return "cpu"
 
 #### Handle versions of napari
 def version_napari_above( compare_version ):