Skip to content

Commit c46439f

Browse files
committed
update cudnn to 8.7.0.84 for CUDA 11.8 builds
1 parent 62103bf commit c46439f

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

common/install_cuda.sh

+4-4
Original file line numberDiff line numberDiff line change
@@ -66,10 +66,10 @@ function install_118 {
6666

6767
# cuDNN license: https://developer.nvidia.com/cudnn/license_agreement
6868
mkdir tmp_cudnn && cd tmp_cudnn
69-
wget -q https://ossci-linux.s3.amazonaws.com/cudnn-linux-x86_64-8.5.0.96_cuda11-archive.tar.xz -O cudnn-linux-x86_64-8.5.0.96_cuda11-archive.tar.xz
70-
tar xf cudnn-linux-x86_64-8.5.0.96_cuda11-archive.tar.xz
71-
cp -a cudnn-linux-x86_64-8.5.0.96_cuda11-archive/include/* /usr/local/cuda/include/
72-
cp -a cudnn-linux-x86_64-8.5.0.96_cuda11-archive/lib/* /usr/local/cuda/lib64/
69+
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
70+
tar xf cudnn-linux-x86_64-8.7.0.84_cuda11-archive.tar.xz
71+
cp -a cudnn-linux-x86_64-8.7.0.84_cuda11-archive/include/* /usr/local/cuda/include/
72+
cp -a cudnn-linux-x86_64-8.7.0.84_cuda11-archive/lib/* /usr/local/cuda/lib64/
7373
cd ..
7474
rm -rf tmp_cudnn
7575
ldconfig

conda/pytorch-nightly/build.sh

+1-1
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@ if [[ -n "$build_with_cuda" ]]; then
7171
DEPS_LIST=(/usr/local/cuda/lib64/libcudnn*.so.8 /usr/local/cuda-11.7/extras/CUPTI/lib64/libcupti.so.11.7)
7272
elif [[ $CUDA_VERSION == 11.8* ]]; then
7373
export TORCH_CUDA_ARCH_LIST="$TORCH_CUDA_ARCH_LIST;6.0;6.1;7.0;7.5;8.0;8.6"
74-
#for cuda 11.8 we use cudnn 8.5
74+
#for cuda 11.8 we use cudnn 8.7
7575
#which does not have single static libcudnn_static.a deliverable to link with
7676
export USE_STATIC_CUDNN=0
7777
#for cuda 11.8 include all dynamic loading libraries

0 commit comments

Comments
 (0)