From d6ca199e3ef606ffc33bd01dcad3547371ebc74d Mon Sep 17 00:00:00 2001
From: Xinle Sheila Liu <sliuxl@amazon.com>
Date: Wed, 28 Jun 2023 17:23:11 -0700
Subject: [PATCH 1/2] Initial copy of install-cudnn-8.3.2 to
 install-cudnn-8.5.0

---
 docker/ci-wheel/install-cudnn-8.5.0 | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)
 create mode 100755 docker/ci-wheel/install-cudnn-8.5.0

diff --git a/docker/ci-wheel/install-cudnn-8.5.0 b/docker/ci-wheel/install-cudnn-8.5.0
new file mode 100755
index 0000000..2d8ca04
--- /dev/null
+++ b/docker/ci-wheel/install-cudnn-8.5.0
@@ -0,0 +1,26 @@
+#!/usr/bin/env bash
+
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the BSD-style license found in the
+# LICENSE file in the root directory of this source tree.
+
+set -o errexit
+
+# cuDNN license: https://developer.nvidia.com/cudnn/license_agreement
+
+mkdir cudnn && cd cudnn
+
+# Taken from https://github.com/pytorch/builder/blob/main/common/install_cuda.sh.
+curl --location --fail --output cudnn.tar.xz\
+    https://developer.download.nvidia.com/compute/redist/cudnn/v8.3.2/local_installers/11.5/cudnn-linux-x86_64-8.3.2.44_cuda11.5-archive.tar.xz
+
+tar xf cudnn.tar.xz
+
+cp cudnn-linux-x86_64-8.3.2.44_cuda11.5-archive/include/* /usr/local/cuda/include
+cp cudnn-linux-x86_64-8.3.2.44_cuda11.5-archive/lib/* /usr/local/cuda/lib64
+
+cd ..
+
+rm -rf cudnn

From 0093fca5d794886ea26b9af1ea27d4bd0df1bfd1 Mon Sep 17 00:00:00 2001
From: Xinle Sheila Liu <sliuxl@amazon.com>
Date: Wed, 28 Jun 2023 17:25:05 -0700
Subject: [PATCH 2/2] Minor changes to install CUDNN with CUDA 11.7

---
 docker/ci-wheel/install-cudnn-8.5.0 | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/docker/ci-wheel/install-cudnn-8.5.0 b/docker/ci-wheel/install-cudnn-8.5.0
index 2d8ca04..094b319 100755
--- a/docker/ci-wheel/install-cudnn-8.5.0
+++ b/docker/ci-wheel/install-cudnn-8.5.0
@@ -10,16 +10,19 @@ set -o errexit
 
 # cuDNN license: https://developer.nvidia.com/cudnn/license_agreement
 
-mkdir cudnn && cd cudnn
+rm -rf cudnn && mkdir cudnn && cd cudnn
 
 # Taken from https://github.com/pytorch/builder/blob/main/common/install_cuda.sh.
 curl --location --fail --output cudnn.tar.xz\
-    https://developer.download.nvidia.com/compute/redist/cudnn/v8.3.2/local_installers/11.5/cudnn-linux-x86_64-8.3.2.44_cuda11.5-archive.tar.xz
+    https://developer.download.nvidia.com/compute/redist/cudnn/v8.5.0/local_installers/11.7/cudnn-linux-x86_64-8.5.0.96_cuda11-archive.tar.xz
 
 tar xf cudnn.tar.xz
 
-cp cudnn-linux-x86_64-8.3.2.44_cuda11.5-archive/include/* /usr/local/cuda/include
-cp cudnn-linux-x86_64-8.3.2.44_cuda11.5-archive/lib/* /usr/local/cuda/lib64
+CUDA_DIR=/usr/local/cuda
+WORK_DIR=`ls | grep "cudnn\-linux-"`  # cudnn-linux-x86_64-8.5.0.96_cuda11-archive
+echo $WORR_DIR
+cp $WORK_DIR/include/* $CUDA_DIR/include
+cp $WORK_DIR/lib/* $CUDA_DIR/lib64
 
 cd ..