From 8cb40599d2a24f01d917e5517d5b957b01608c63 Mon Sep 17 00:00:00 2001
From: Ting Lu <tingl@nvidia.com>
Date: Fri, 15 Aug 2025 11:42:31 -0700
Subject: [PATCH 1/2] Update CUDNN to 9.12.0.46

---
 aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1 | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1 b/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1
index 7378fa5274..9423eb7e00 100644
--- a/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1
+++ b/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1
@@ -15,7 +15,7 @@ $installerArgs = "nvcc_$cudaVersion cuobjdump_$cudaVersion nvprune_$cudaVersion
 # Switch statement for specfic CUDA versions
 $cudnn_subfolder="cuda"
 $cudnn_lib_folder="lib\x64"
-$cudnn_subfolder="cudnn-windows-x86_64-9.10.2.21_cuda12-archive"
+$cudnn_subfolder="cudnn-windows-x86_64-9.12.0.46_cuda12-archive"
 $toolkitInstaller = "cuda_12.6.3_561.17_windows.exe"
 
 Switch ($cudaVersion) {
@@ -32,6 +32,7 @@ Switch ($cudaVersion) {
     $installerArgs += " cuda_profiler_api_$cudaVersion nvjitlink_$cudaVersion"
   }
   "13.0" {
+    $cudnn_subfolder="cudnn-windows-x86_64-9.12.0.46_cuda13-archive"
     $toolkitInstaller = "cuda_13.0.0_windows.exe"
     $installerArgs += " cuda_profiler_api_$cudaVersion nvjitlink_$cudaVersion"
   }

From 317af2fde960d18fc96007c2b70ec08e24311eff Mon Sep 17 00:00:00 2001
From: Ting Lu <tingl@nvidia.com>
Date: Fri, 15 Aug 2025 14:12:10 -0700
Subject: [PATCH 2/2] do not change cuda 12 cudnn ver yet

---
 aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1 | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1 b/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1
index 9423eb7e00..db2bdb584d 100644
--- a/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1
+++ b/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1
@@ -15,7 +15,7 @@ $installerArgs = "nvcc_$cudaVersion cuobjdump_$cudaVersion nvprune_$cudaVersion
 # Switch statement for specfic CUDA versions
 $cudnn_subfolder="cuda"
 $cudnn_lib_folder="lib\x64"
-$cudnn_subfolder="cudnn-windows-x86_64-9.12.0.46_cuda12-archive"
+$cudnn_subfolder="cudnn-windows-x86_64-9.10.2.21_cuda12-archive"
 $toolkitInstaller = "cuda_12.6.3_561.17_windows.exe"
 
 Switch ($cudaVersion) {