Update the default version in cuda utils

pytorch · Jun 12, 2024 · 1d2830a · 1d2830a
1 parent 89f5e40
commit 1d2830a
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/utils/cuda_utils.py b/utils/cuda_utils.py
@@ -8,16 +8,12 @@
 from typing import Optional
 
 # defines the default CUDA version to compile against
-DEFAULT_CUDA_VERSION = "12.1"
+DEFAULT_CUDA_VERSION = "12.4"
 
 CUDA_VERSION_MAP = {
-    "11.8": {
-        "pytorch_url": "cu118",
-        "magma_version": "magma-cuda118",
-    },
-    "12.1": {
-        "pytorch_url": "cu121",
-        "magma_version": "magma-cuda121",
+    "12.4": {
+        "pytorch_url": "cu124",
+        "magma_version": "magma-cuda124",
     },
 }
 PIN_CMAKE_VERSION = "3.22.*"