Fix Mac install

dnhkng · Jan 25, 2025 · 7c55b17 · 7c55b17
1 parent ac3f1e3
commit 7c55b17
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/scripts/install.py b/scripts/install.py
@@ -74,7 +74,11 @@ def main() -> None:
         venv_python = ".venv/bin/python"
         os.environ["PATH"] = f"{os.path.dirname(venv_python)}:{os.environ['PATH']}"
 
-    has_cuda = subprocess.run(["nvcc", "--version"], capture_output=True).returncode == 0
+    try:
+        has_cuda = subprocess.run(["nvcc", "--version"], capture_output=True, check=False).returncode == 0
+    except FileNotFoundError:
+        has_cuda = False
+
     extras = "[cuda]" if has_cuda else "[cpu]"
 
     # Install project in editable mode