Add PyTorch 2.4.0

MiroPsota · Jul 25, 2024 · bcbbdbb · bcbbdbb
1 parent d382177
commit bcbbdbb
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 0 deletions.
diff --git a/.github/workflows/cuda/Linux_env.sh b/.github/workflows/cuda/Linux_env.sh
@@ -18,6 +18,7 @@ declare -A CUDA_ARCHS=(
   ["cu118"]="3.7;5.0;6.0;7.0;7.5;8.0;8.6;9.0"
 
   ["cu121"]="5.0;6.0;7.0;7.5;8.0;8.6;9.0"
+  ["cu124"]="5.0;6.0;7.0;7.5;8.0;8.6;9.0"
 )
 export TORCH_CUDA_ARCH_LIST=${CUDA_ARCHS[$CUDA_VERSION]}
 export FORCE_CUDA=1
diff --git a/.github/workflows/cuda/Windows_env.sh b/.github/workflows/cuda/Windows_env.sh
@@ -15,6 +15,7 @@ declare -A CUDA_ARCHS=(
   ["cu118"]="3.7+PTX;5.0;6.0;6.1;7.0;7.5;8.0;8.6;9.0"
 
   ["cu121"]="5.0;6.0;6.1;7.0;7.5;8.0;8.6;9.0"
+  ["cu124"]="5.0;6.0;6.1;7.0;7.5;8.0;8.6;9.0"
 )
 export TORCH_CUDA_ARCH_LIST=${CUDA_ARCHS[$CUDA_VERSION]}
 export FORCE_CUDA=1
diff --git a/.github/workflows/cuda/Windows_install.sh b/.github/workflows/cuda/Windows_install.sh
@@ -16,6 +16,9 @@ declare -A CUDA_LINKS=(
   ["cu120"]="https://developer.download.nvidia.com/compute/cuda/12.0.1/local_installers/cuda_12.0.1_528.33_windows.exe"
   ["cu121"]="https://developer.download.nvidia.com/compute/cuda/12.1.1/local_installers/cuda_12.1.1_531.14_windows.exe"
   ["cu122"]="https://developer.download.nvidia.com/compute/cuda/12.2.2/local_installers/cuda_12.2.2_537.13_windows.exe"
+  ["cu123"]="https://developer.download.nvidia.com/compute/cuda/12.3.2/local_installers/cuda_12.3.2_546.12_windows.exe"
+  ["cu124"]="https://developer.download.nvidia.com/compute/cuda/12.4.1/local_installers/cuda_12.4.1_551.78_windows.exe"
+  ["cu125"]="https://developer.download.nvidia.com/compute/cuda/12.5.1/local_installers/cuda_12.5.1_555.85_windows.exe"
 )
 CUDA_LINK=${CUDA_LINKS[$CUDA_VERSION]}
 

diff --git a/get_jobs.py b/get_jobs.py
@@ -31,6 +31,7 @@ class TorchRelease:
     "2.2.2": TorchRelease(("3.8", "3.9", "3.10", "3.11", "3.12"), ("cpu", "cu118", "cu121", "rocm5.7")),
     "2.3.0": TorchRelease(("3.8", "3.9", "3.10", "3.11", "3.12"), ("cpu", "cu118", "cu121", "rocm6.0")),
     "2.3.1": TorchRelease(("3.8", "3.9", "3.10", "3.11", "3.12"), ("cpu", "cu118", "cu121", "rocm6.0")),
+    "2.4.0": TorchRelease(("3.8", "3.9", "3.10", "3.11", "3.12"), ("cpu", "cu118", "cu121", "cu124", "rocm6.1")),
 }