File tree 1 file changed +4
-1
lines changed
1 file changed +4
-1
lines changed Original file line number Diff line number Diff line change @@ -7,8 +7,11 @@ if [[ ${CUDNN_VERSION} == 8 ]]; then
7
7
if [[ ${CUDA_VERSION: 0: 4} == " 11.7" ]]; then
8
8
CUDNN_NAME=" cudnn-linux-x86_64-8.5.0.96_cuda11-archive"
9
9
curl --retry 3 -OLs https://ossci-linux.s3.amazonaws.com/${CUDNN_NAME} .tar.xz
10
+ elif [[ ${CUDA_VERSION: 0: 4} == " 11.8" ]]; then
11
+ CUDNN_NAME=" cudnn-linux-x86_64-8.7.0.84_cuda11-archive"
12
+ curl --retry 3 -OLs https://developer.download.nvidia.com/compute/redist/cudnn/v8.7.0/local_installers/11.8/${CUDNN_NAME} .tar.xz
10
13
else
11
- curl --retry 3 -OLs https://developer.download.nvidia.com/compute/redist/cudnn/v8.3.2/local_installers/11.5/${CUDNN_NAME} .tar.xz
14
+ curl --retry 3 -OLs https://developer.download.nvidia.com/compute/redist/cudnn/v8.3.2/local_installers/11.5/${CUDNN_NAME} .tar.xz
12
15
fi
13
16
14
17
tar xf ${CUDNN_NAME} .tar.xz
You can’t perform that action at this time.
0 commit comments