From 97cc8c7ce6e7644f0130d1d8c7e1c0a8229b3547 Mon Sep 17 00:00:00 2001
From: Ashish Karale <akarale@nvidia.com>
Date: Fri, 7 Aug 2026 01:13:47 -0700
Subject: [PATCH 1/2] Add Windows AMI CUDA 13.4

---
 aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1 | 5 +++++
 aws/ami/windows/windows.pkr.hcl                           | 7 +++++++
 2 files changed, 12 insertions(+)

diff --git a/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1 b/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1
index d68fa8e179..920965297d 100644
--- a/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1
+++ b/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1
@@ -38,6 +38,11 @@ Switch ($cudaVersion) {
     $toolkitInstaller = "cuda_13.2.1_windows.exe"
     $installerArgs = ""
   }
+  "13.4" {
+    $cudnn_subfolder="cudnn-windows-x86_64-9.20.0.48_cuda13-archive"
+    $toolkitInstaller = "cuda_13.4.0_windows_x86_64.exe"
+    $installerArgs = ""
+  }
 }
 
 
diff --git a/aws/ami/windows/windows.pkr.hcl b/aws/ami/windows/windows.pkr.hcl
index ec02158a32..862f11a257 100644
--- a/aws/ami/windows/windows.pkr.hcl
+++ b/aws/ami/windows/windows.pkr.hcl
@@ -127,6 +127,13 @@ build {
     ]
   }
 
+  provisioner "powershell" {
+    environment_vars = ["CUDA_VERSION=13.4"]
+    scripts = [
+      "${path.root}/scripts/Installers/Install-CUDA-Tools.ps1",
+    ]
+  }
+
   # Uninstall Windows Defender, it brings more trouble than it's worth. Do this
   # last as it screws up the installation of other services like sshd somehow
   provisioner "powershell" {

From 4a01dceb227dd3c1b78ac6da51feaf77da1d823d Mon Sep 17 00:00:00 2001
From: Ashish Karale <akarale@nvidia.com>
Date: Fri, 21 Aug 2026 11:56:49 +0530
Subject: [PATCH 2/2] Remove cuda 12.9 AMI from the list

---
 aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1 | 3 ---
 aws/ami/windows/windows.pkr.hcl                           | 7 -------
 2 files changed, 10 deletions(-)

diff --git a/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1 b/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1
index 920965297d..ca293ead8a 100644
--- a/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1
+++ b/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1
@@ -25,9 +25,6 @@ Switch ($cudaVersion) {
   "12.8" {
     $toolkitInstaller = "cuda_12.8.1_572.61_windows.exe"
   }
-  "12.9" {
-    $toolkitInstaller = "cuda_12.9.1_576.57_windows.exe"
-  }
   "13.0" {
     $cudnn_subfolder="cudnn-windows-x86_64-9.20.0.48_cuda13-archive"
     $toolkitInstaller = "cuda_13.0.0_windows.exe"
diff --git a/aws/ami/windows/windows.pkr.hcl b/aws/ami/windows/windows.pkr.hcl
index 862f11a257..e732bd6e11 100644
--- a/aws/ami/windows/windows.pkr.hcl
+++ b/aws/ami/windows/windows.pkr.hcl
@@ -106,13 +106,6 @@ build {
     ]
   }
 
-  provisioner "powershell" {
-    environment_vars = ["CUDA_VERSION=12.9"]
-    scripts = [
-      "${path.root}/scripts/Installers/Install-CUDA-Tools.ps1",
-    ]
-  }
-
   provisioner "powershell" {
     environment_vars = ["CUDA_VERSION=13.0"]
     scripts = [