Skip to content

Commit 286cca8

Browse files
atalmanpytorchmergebot
authored andcommitted
Add cudnn install 8.7.0.84 for CUDA 11.8 (#93086)
Add cudnn install 8.7.0.84 for CUDA 11.8 . Same as: #84964 Related to pytorch/builder#1271 Test PR: #92971 Pull Request resolved: #93086 Approved by: https://github.com/kit1980, https://github.com/malfet
1 parent 0ecb071 commit 286cca8

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

.circleci/docker/common/install_cudnn.sh

+4-1
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,11 @@ if [[ ${CUDNN_VERSION} == 8 ]]; then
77
if [[ ${CUDA_VERSION:0:4} == "11.7" ]]; then
88
CUDNN_NAME="cudnn-linux-x86_64-8.5.0.96_cuda11-archive"
99
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
1013
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
1215
fi
1316

1417
tar xf ${CUDNN_NAME}.tar.xz

0 commit comments

Comments
 (0)