8000 Update to CUDNN 9.10 · pytorch/pytorch@1e5d25b · GitHub
[go: up one dir, main page]
Skip to content

Commit 1e5d25b

Browse files
Skylion007Skylion007
committed
Update to CUDNN 9.10
1 parent cf6a4b4 commit 1e5d25b

File tree

7 files changed

+40
-40
lines changed

7 files changed

+40
-40
lines changed

‎.ci/docker/common/install_cuda.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,7 @@ function install_124 {
6969
}
7070

7171
function install_126 {
72-
CUDNN_VERSION=9.5.1.17
72+
CUDNN_VERSION=9.10.0.56
7373
echo "Installing CUDA 12.6.3 and cuDNN ${CUDNN_VERSION} and NCCL and cuSparseLt-0.6.3"
7474
install_cuda 12.6.3 cuda_12.6.3_560.35.05_linux
7575

@@ -182,7 +182,7 @@ function prune_126 {
182182
}
183183

184184
function install_128 {
185-
CUDNN_VERSION=9.9.0.52
185+
CUDNN_VERSION=9.10.0.56
186186
echo "Installing CUDA 12.8.0 and cuDNN ${CUDNN_VERSION} and NCCL and cuSparseLt-0.6.3"
187187
# install CUDA 12.8.0 in the same container
188188
install_cuda 12.8.0 cuda_12.8.0_570.86.10_linux

‎.ci/docker/common/install_cudnn.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,9 +5,9 @@ if [[ -n "${CUDNN_VERSION}" ]]; then
55
mkdir tmp_cudnn
66
pushd tmp_cudnn
77
if [[ ${CUDA_VERSION:0:4} == "12.8" ]]; then
8-
CUDNN_NAME="cudnn-linux-x86_64-9.9.0.52_cuda12-archive"
8+
CUDNN_NAME="cudnn-linux-x86_64-9.10.0.56_cuda12-archive"
99
elif [[ ${CUDA_VERSION:0:4} == "12.6" ]]; then
10-
CUDNN_NAME="cudnn-linux-x86_64-9.9.0.52_cuda12-archive"
10+
CUDNN_NAME="cudnn-linux-x86_64-9.10.0.56_cuda12-archive"
1111
elif [[ ${CUDA_VERSION:0:2} == "12" ]]; then
1212
CUDNN_NAME="cudnn-linux-x86_64-9.1.0.70_cuda12-archive"
1313
elif [[ ${CUDA_VERSION:0:2} == "11" ]]; then

‎.github/scripts/generate_binary_build_matrix.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@
5959
"nvidia-cuda-nvrtc-cu12==12.6.77; platform_system == 'Linux' and platform_machine == 'x86_64' | "
6060
"nvidia-cuda-runtime-cu12==12.6.77; platform_system == 'Linux' and platform_machine == 'x86_64' | "
6161
"nvidia-cuda-cupti-cu12==12.6.80; platform_system == 'Linux' and platform_machine == 'x86_64' | "
62-
"nvidia-cudnn-cu12==9.9.0.52; platform_system == 'Linux' and platform_machine == 'x86_64' | "
62+
"nvidia-cudnn-cu12==9.10.0.56; platform_system == 'Linux' and platform_machine == 'x86_64' | "
6363
"nvidia-cublas-cu12==12.6.4.1; platform_system == 'Linux' and platform_machine == 'x86_64' | "
6464
"nvidia-cufft-cu12==11.3.0.4; platform_system == 'Linux' and platform_machine == 'x86_64' | "
6565
"nvidia-curand-cu12==10.3.7.77; platform_system == 'Linux' and platform_machine == 'x86_64' | "
@@ -75,7 +75,7 @@
7575
"nvidia-cuda-nvrtc-cu12==12.8.61; platform_system == 'Linux' and platform_machine == 'x86_64' | "
7676
"nvidia-cuda-runtime-cu12==12.8.57; platform_system == 'Linux' and platform_machine == 'x86_64' | "
7777
"nvidia-cuda-cupti-cu12==12.8.57; platform_system == 'Linux' and platform_machine == 'x86_64' | "
78-
"nvidia-cudnn-cu12==9.9.0.52; platform_system == 'Linux' and platform_machine == 'x86_64' | "
78+
"nvidia-cudnn-cu12==9.10.0.56; platform_system == 'Linux' and platform_machine == 'x86_64' | "
7979
"nvidia-cublas-cu12==12.8.3.14; platform_system == 'Linux' and platform_machine == 'x86_64' | "
8080
"nvidia-cufft-cu12==11.3.3.41; platform_system == 'Linux' and platform_machine == 'x86_64' | "
8181
"nvidia-curand-cu12==10.3.9.55; platform_system == 'Linux' and platform_machine == 'x86_64' | "

‎.github/workflows/generated-linux-aarch64-binary-manywheel-nightly.yml

Lines changed: 6 additions & 6 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

‎.github/workflows/generated-linux-binary-manywheel-main.yml

Lines changed: 2 additions & 2 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)
0