Skip to content

Commit 71e4a2a

Browse files
authored
Update cudnn versions to 8.9.7 (#1715)
For both 12.1 and 11.8 CUDA Companion change for pytorch/pytorch#120642
1 parent cfc2d22 commit 71e4a2a

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

common/install_cuda.sh

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -36,10 +36,10 @@ function install_118 {
3636

3737
# cuDNN license: https://developer.nvidia.com/cudnn/license_agreement
3838
mkdir tmp_cudnn && cd tmp_cudnn
39-
wget -q https://developer.download.nvidia.com/compute/redist/cudnn/v8.7.0/local_installers/11.8/cudnn-linux-x86_64-8.7.0.84_cuda11-archive.tar.xz -O cudnn-linux-x86_64-8.7.0.84_cuda11-archive.tar.xz
40-
tar xf cudnn-linux-x86_64-8.7.0.84_cuda11-archive.tar.xz
41-
cp -a cudnn-linux-x86_64-8.7.0.84_cuda11-archive/include/* /usr/local/cuda/include/
42-
cp -a cudnn-linux-x86_64-8.7.0.84_cuda11-archive/lib/* /usr/local/cuda/lib64/
39+
wget -q https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/linux-x86_64/cudnn-linux-x86_64-8.9.7.29_cuda11-archive.tar.xz
40+
tar xf cudnn-linux-x86_64-8.9.7.29_cuda11-archive.tar.xz
41+
cp -a cudnn-linux-x86_64-8.9.7.29_cuda11-archive/include/* /usr/local/cuda/include/
42+
cp -a cudnn-linux-x86_64-8.9.7.29_cuda11-archive/lib/* /usr/local/cuda/lib64/
4343
cd ..
4444
rm -rf tmp_cudnn
4545

@@ -69,10 +69,10 @@ function install_121 {
6969

7070
# cuDNN license: https://developer.nvidia.com/cudnn/license_agreement
7171
mkdir tmp_cudnn && cd tmp_cudnn
72-
wget -q https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/linux-x86_64/cudnn-linux-x86_64-8.9.2.26_cuda12-archive.tar.xz -O cudnn-linux-x86_64-8.9.2.26_cuda12-archive.tar.xz
73-
tar xf cudnn-linux-x86_64-8.9.2.26_cuda12-archive.tar.xz
74-
cp -a cudnn-linux-x86_64-8.9.2.26_cuda12-archive/include/* /usr/local/cuda/include/
75-
cp -a cudnn-linux-x86_64-8.9.2.26_cuda12-archive/lib/* /usr/local/cuda/lib64/
72+
wget -q https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/linux-x86_64/cudnn-linux-x86_64-8.9.7.29_cuda12-archive.tar.xz -O cudnn-linux-x86_64-8.9.7.29_cuda12-archive.tar.xz
73+
tar xf cudnn-linux-x86_64-8.9.7.29_cuda12-archive.tar.xz
74+
cp -a cudnn-linux-x86_64-8.9.7.29_cuda12-archive/include/* /usr/local/cuda/include/
75+
cp -a cudnn-linux-x86_64-8.9.7.29_cuda12-archive/lib/* /usr/local/cuda/lib64/
7676
cd ..
7777
rm -rf tmp_cudnn
7878

0 commit comments

Comments
 (0)