diff --git a/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1 b/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1 index d68fa8e179..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" @@ -38,6 +35,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..e732bd6e11 100644 --- a/aws/ami/windows/windows.pkr.hcl +++ b/aws/ami/windows/windows.pkr.hcl @@ -107,21 +107,21 @@ build { } provisioner "powershell" { - environment_vars = ["CUDA_VERSION=12.9"] + environment_vars = ["CUDA_VERSION=13.0"] scripts = [ "${path.root}/scripts/Installers/Install-CUDA-Tools.ps1", ] } provisioner "powershell" { - environment_vars = ["CUDA_VERSION=13.0"] + environment_vars = ["CUDA_VERSION=13.2"] scripts = [ "${path.root}/scripts/Installers/Install-CUDA-Tools.ps1", ] } provisioner "powershell" { - environment_vars = ["CUDA_VERSION=13.2"] + environment_vars = ["CUDA_VERSION=13.4"] scripts = [ "${path.root}/scripts/Installers/Install-CUDA-Tools.ps1", ]