OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

CUDNN changes for cuda 11.5 (#71869)

Browse Source 
Summary:
CUDNN changes for cuda 11.5

Pull Request resolved: https://github.com/pytorch/pytorch/pull/71869

Reviewed By: janeyx99

Differential Revision: D33817943

Pulled By: atalman

fbshipit-source-id: 5da5f8f45877ac12c0ee4d982082fd24e5f09adb
(cherry picked from commit 3f3d96af69)
This commit is contained in:
Andrey Talman 2022-01-27 07:26:20 -08:00 committed by PyTorch MergeBot
parent b66f1bc80f
commit 21d307cd22
2 changed files with 16 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.circleci/docker/common/install_cudnn8.sh
View File

@ -7,4 +7,4 @@ sudo apt-get update
# that could not be found:
# plm_rsh_agent: ssh : rsh
sudo apt-get install -y ssh
sudo apt-get update && apt-get install -y --no-install-recommends libcudnn8=8.2.0.53-1+cuda11.3 libcudnn8-dev=8.2.0.53-1+cuda11.3 && apt-mark hold libcudnn8
sudo apt-get update && apt-get install -y --no-install-recommends libcudnn8=8.3.2.44-1+cuda11.5 libcudnn8-dev=8.3.2.44-1+cuda11.5 && apt-mark hold libcudnn8

25
.circleci/scripts/windows_cudnn_install.sh
View File

@ -1,26 +1,26 @@
#!/bin/bash
set -eux -o pipefail
# This is typically blank but for CUDA 10* it'll be set to 10
windows_version_qualifier=""
windows_s3_link="https://ossci-windows.s3.amazonaws.com"
case ${CUDA_VERSION} in
10.1)
archive_version="v7.6.4.38"
windows_version_qualifier="10"
# This is typically blank but for CUDA 10* it'll be set to 10
cudnn_file_name="cudnn-${CUDA_VERSION}-windows10-x64-v7.6.4.38"
;;
10.2)
archive_version="v7.6.5.32"
windows_version_qualifier="10"
cudnn_file_name="cudnn-${CUDA_VERSION}-windows10-x64-v7.6.5.32"
;;
11.1)
archive_version="v8.0.5.39"
cudnn_file_name="cudnn-${CUDA_VERSION}-windows-x64-v8.0.5.39"
;;
11.3)
archive_version="v8.2.0.53"
cudnn_file_name="cudnn-${CUDA_VERSION}-windows-x64-v8.2.0.53"
;;
11.5)
archive_version="v8.2.0.53"
# Since cudnn 8.3 the filename have changed
cudnn_file_name="cudnn-windows-x86_64-8.3.2.44_cuda${CUDA_VERSION}-archive"
;;
*)
echo "CUDA_VERSION: ${CUDA_VERSION} not supported yet"
@ -29,7 +29,7 @@ case ${CUDA_VERSION} in
esac
cudnn_installer_name="cudnn_installer.zip"
cudnn_installer_link="https://ossci-windows.s3.amazonaws.com/cudnn-${CUDA_VERSION}-windows${windows_version_qualifier}-x64-${archive_version}.zip"
cudnn_installer_link="${windows_s3_link}/${cudnn_file_name}.zip"
cudnn_install_folder="C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v${CUDA_VERSION}/"
if [[ -f "${cudnn_install_folder}/include/cudnn.h" ]]; then
@ -44,6 +44,11 @@ else
# Remove all of the directories before attempting to copy files
rm -rf "${cudnn_install_folder:?}/*"
cp -rf cudnn/cuda/* "${cudnn_install_folder}"
#Make sure windows path contains zlib dll
curl -k -L "${windows_s3_link}/zlib123dllx64.zip" --output "${tmp_dir}\zlib123dllx64.zip"
7z x "${tmp_dir}\zlib123dllx64.zip" -o"${tmp_dir}\zlib"
xcopy /Y "${tmp_dir}\zlib\dll_x64\*.dll" "C:\Windows\System32"
)
rm -rf "${tmp_dir}"
fi