diff --git a/.circleci/docker/common/install_cudnn.sh b/.circleci/docker/common/install_cudnn.sh index 1f1c34ea200d4..4a8829b1cba11 100644 --- a/.circleci/docker/common/install_cudnn.sh +++ b/.circleci/docker/common/install_cudnn.sh @@ -4,7 +4,13 @@ if [[ ${CUDNN_VERSION} == 8 ]]; then # cuDNN license: https://developer.nvidia.com/cudnn/license_agreement mkdir tmp_cudnn && cd tmp_cudnn CUDNN_NAME="cudnn-linux-x86_64-8.3.2.44_cuda11.5-archive" - curl -OLs https://developer.download.nvidia.com/compute/redist/cudnn/v8.3.2/local_installers/11.5/${CUDNN_NAME}.tar.xz + if [[ ${CUDA_VERSION:0:4} == "11.7" ]]; then + CUDNN_NAME="cudnn-linux-x86_64-8.5.0.96_cuda11-archive" + curl -OLs https://ossci-linux.s3.amazonaws.com/${CUDNN_NAME}.tar.xz + else + curl -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 cp -a ${CUDNN_NAME}/include/* /usr/include/ cp -a ${CUDNN_NAME}/include/* /usr/local/cuda/include/ diff --git a/.circleci/docker/ubuntu-cuda/Dockerfile b/.circleci/docker/ubuntu-cuda/Dockerfile index 4375b612a308b..53349bfec5334 100644 --- a/.circleci/docker/ubuntu-cuda/Dockerfile +++ b/.circleci/docker/ubuntu-cuda/Dockerfile @@ -118,6 +118,7 @@ COPY --from=pytorch/llvm:9.0.1 /opt/llvm /opt/llvm # Install CUDNN ARG CUDNN_VERSION +ARG CUDA_VERSION COPY ./common/install_cudnn.sh install_cudnn.sh RUN if [ "${CUDNN_VERSION}" -eq 8 ]; then bash install_cudnn.sh; fi RUN rm install_cudnn.sh diff --git a/.circleci/scripts/windows_cudnn_install.sh b/.circleci/scripts/windows_cudnn_install.sh index 763bc950fc4be..c279259e83416 100644 --- a/.circleci/scripts/windows_cudnn_install.sh +++ b/.circleci/scripts/windows_cudnn_install.sh @@ -18,7 +18,7 @@ case ${CUDA_VERSION} in ;; 11.7) # Use cudnn8.3 with hard-coded cuda11.5 version - cudnn_file_name="cudnn-windows-x86_64-8.3.2.44_cuda11.5-archive" + cudnn_file_name="cudnn-windows-x86_64-8.5.0.96_cuda11-archive" ;; *) echo "CUDA_VERSION: ${CUDA_VERSION} not supported yet"