OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2026-01-15 12:15:51 +00:00
Code Issues Actions 51 Packages Projects Releases Wiki Activity

Add cudnn install 8.7.0.84 for CUDA 11.8 (#93086)

Browse Source 
Add cudnn install 8.7.0.84 for CUDA 11.8 .

Same as: https://github.com/pytorch/pytorch/pull/84964
Related to https://github.com/pytorch/builder/pull/1271
Test PR: https://github.com/pytorch/pytorch/pull/92971
Pull Request resolved: https://github.com/pytorch/pytorch/pull/93086
Approved by: https://github.com/kit1980, https://github.com/malfet
This commit is contained in:
atalman
2023-01-30 22:53:20 +00:00
committed by PyTorch MergeBot
parent 0ecb071fc4
commit 286cca8929
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
.circleci/docker/common/install_cudnn.sh
View File

@@ -7,8 +7,11 @@ if [[ ${CUDNN_VERSION} == 8 ]]; then
if [[ ${CUDA_VERSION:0:4} == "11.7" ]]; then
CUDNN_NAME="cudnn-linux-x86_64-8.5.0.96_cuda11-archive"
curl --retry 3 -OLs https://ossci-linux.s3.amazonaws.com/${CUDNN_NAME}.tar.xz
elif [[ ${CUDA_VERSION:0:4} == "11.8" ]]; then
CUDNN_NAME="cudnn-linux-x86_64-8.7.0.84_cuda11-archive"
curl --retry 3 -OLs https://developer.download.nvidia.com/compute/redist/cudnn/v8.7.0/local_installers/11.8/${CUDNN_NAME}.tar.xz
else
curl --retry 3 -OLs https://developer.download.nvidia.com/compute/redist/cudnn/v8.3.2/local_installers/11.5/${CUDNN_NAME}.tar.xz
curl --retry 3 -OLs https://developer.download.nvidia.com/compute/redist/cudnn/v8.3.2/local_installers/11.5/${CUDNN_NAME}.tar.xz
fi
tar xf ${CUDNN_NAME}.tar.xz