diff --git a/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1 b/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1 index a460158fa0..7378fa5274 100644 --- a/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1 +++ b/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1 @@ -31,6 +31,10 @@ Switch ($cudaVersion) { $toolkitInstaller = "cuda_12.9.1_576.57_windows.exe" $installerArgs += " cuda_profiler_api_$cudaVersion nvjitlink_$cudaVersion" } + "13.0" { + $toolkitInstaller = "cuda_13.0.0_windows.exe" + $installerArgs += " cuda_profiler_api_$cudaVersion nvjitlink_$cudaVersion" + } } diff --git a/aws/ami/windows/windows.pkr.hcl b/aws/ami/windows/windows.pkr.hcl index dd4247fab8..a02bc69ca1 100644 --- a/aws/ami/windows/windows.pkr.hcl +++ b/aws/ami/windows/windows.pkr.hcl @@ -111,6 +111,13 @@ build { ] } + provisioner "powershell" { + environment_vars = ["CUDA_VERSION=13.0"] + 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" {