diff --git a/.github/workflows/cuda/Linux_env.sh b/.github/workflows/cuda/Linux_env.sh index 0a41ffc..2b3cccb 100644 --- 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 index 3ef5a7e..c77d268 100644 --- 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 index 068926d..1eef757 100644 --- 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 index dc89a40..0a59a04 100644 --- 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")), }