Skip to content

Commit 1d2830a

Browse files
committed
Update the default version in cuda utils
1 parent 89f5e40 commit 1d2830a

File tree

1 file changed

+4
-8
lines changed

1 file changed

+4
-8
lines changed

utils/cuda_utils.py

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -8,16 +8,12 @@
88
from typing import Optional
99

1010
# defines the default CUDA version to compile against
11-
DEFAULT_CUDA_VERSION = "12.1"
11+
DEFAULT_CUDA_VERSION = "12.4"
1212

1313
CUDA_VERSION_MAP = {
14-
"11.8": {
15-
"pytorch_url": "cu118",
16-
"magma_version": "magma-cuda118",
17-
},
18-
"12.1": {
19-
"pytorch_url": "cu121",
20-
"magma_version": "magma-cuda121",
14+
"12.4": {
15+
"pytorch_url": "cu124",
16+
"magma_version": "magma-cuda124",
2117
},
2218
}
2319
PIN_CMAKE_VERSION = "3.22.*"

0 commit comments

Comments
 (0)