File tree 2 files changed +5
-5
lines changed
2 files changed +5
-5
lines changed Original file line number Diff line number Diff line change @@ -66,10 +66,10 @@ function install_118 {
66
66
67
67
# cuDNN license: https://developer.nvidia.com/cudnn/license_agreement
68
68
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/
73
73
cd ..
74
74
rm -rf tmp_cudnn
75
75
ldconfig
Original file line number Diff line number Diff line change @@ -71,7 +71,7 @@ if [[ -n "$build_with_cuda" ]]; then
71
71
DEPS_LIST=(/usr/local/cuda/lib64/libcudnn* .so.8 /usr/local/cuda-11.7/extras/CUPTI/lib64/libcupti.so.11.7)
72
72
elif [[ $CUDA_VERSION == 11.8* ]]; then
73
73
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
75
75
# which does not have single static libcudnn_static.a deliverable to link with
76
76
export USE_STATIC_CUDNN=0
77
77
# for cuda 11.8 include all dynamic loading libraries
You can’t perform that action at this time.
0 commit comments