8000 [BE]: Update cudnn to 9.9 for cu128 · pytorch/pytorch@c9d09a0 · GitHub
[go: up one dir, main page]
Skip to content

Commit c9d09a0

Browse files
Skylion007Skylion007
committed
[BE]: Update cudnn to 9.9 for cu128
1 parent 8f54e56 commit c9d09a0

6 files changed

+25
-25
lines changed

‎.ci/docker/common/install_cuda.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -182,7 +182,7 @@ function prune_126 {
182182
}
183183

184184
function install_128 {
185-
CUDNN_VERSION=9.8.0.87
185+
CUDNN_VERSION=9.9.0.52
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.8.0.87_cuda12-archive"
8+
CUDNN_NAME="cudnn-linux-x86_64-9.9.0.52_cuda12-archive"
99
elif [[ ${CUDA_VERSION:0:4} == "12.6" ]]; then
10-
CUDNN_NAME="cudnn-linux-x86_64-9.5.1.17_cuda12-archive"
10+
CUDNN_NAME="cudnn-linux-x86_64-9.9.0.52_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.5.1.17; platform_system == 'Linux' and platform_machine == 'x86_64' | "
62+
"nvidia-cudnn-cu12==9.9.0.52; 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.8.0.87; platform_system == 'Linux' and platform_machine == 'x86_64' | "
78+
"nvidia-cudnn-cu12==9.9.0.52; 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.

0 commit comments

Comments
 (0)
0