Revert "no defaults for WITH_CUDA envvar. default is autodetection"

This reverts commit f751306.
torchmd · Feb 14, 2025 · 6de4a2e · 6de4a2e
1 parent f751306
commit 6de4a2e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -9,9 +9,9 @@
 
 
 # If WITH_CUDA is defined
-if os.environ.get("WITH_CUDA") == "1":
+if os.environ.get("WITH_CUDA", "0") == "1":
     use_cuda = True
-elif os.environ.get("WITH_CUDA") == "0":
+elif os.environ.get("WITH_CUDA", "0") == "0":
     use_cuda = False
 else:
     use_cuda = torch.cuda._is_compiled()