Skip to content

Commit 304ebf2

Browse files
authored
Add Windows AMI CUDA 13.4 (#8465)
### Summary Adds CUDA 13.4 to the Windows AMI build. - Adds CUDA 13.4 toolkit and cuDNN configuration to Install-CUDA-Tools.ps1 - Provisions CUDA 13.4 in windows.pkr.hcl ### Validation Ran packer validate Ran packer fmt -check ### Follow-up [cuda_13.4.0_windows_x86_64.exe](https://packages.nvidia.com/prerelease/cuda/13.4.0/local_installers/cuda_13.4.0_windows_x86_64.exe) must be available in the ossci-windows S3 bucket before the AMI build can succeed. CC: @atalman @tinglvv @ptrblck
1 parent 2f89f62 commit 304ebf2

2 files changed

Lines changed: 8 additions & 6 deletions

File tree

aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -25,9 +25,6 @@ Switch ($cudaVersion) {
2525
"12.8" {
2626
$toolkitInstaller = "cuda_12.8.1_572.61_windows.exe"
2727
}
28-
"12.9" {
29-
$toolkitInstaller = "cuda_12.9.1_576.57_windows.exe"
30-
}
3128
"13.0" {
3229
$cudnn_subfolder="cudnn-windows-x86_64-9.20.0.48_cuda13-archive"
3330
$toolkitInstaller = "cuda_13.0.0_windows.exe"
@@ -38,6 +35,11 @@ Switch ($cudaVersion) {
3835
$toolkitInstaller = "cuda_13.2.1_windows.exe"
3936
$installerArgs = ""
4037
}
38+
"13.4" {
39+
$cudnn_subfolder="cudnn-windows-x86_64-9.20.0.48_cuda13-archive"
40+
$toolkitInstaller = "cuda_13.4.0_windows_x86_64.exe"
41+
$installerArgs = ""
42+
}
4143
}
4244

4345

aws/ami/windows/windows.pkr.hcl

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -107,21 +107,21 @@ build {
107107
}
108108

109109
provisioner "powershell" {
110-
environment_vars = ["CUDA_VERSION=12.9"]
110+
environment_vars = ["CUDA_VERSION=13.0"]
111111
scripts = [
112112
"${path.root}/scripts/Installers/Install-CUDA-Tools.ps1",
113113
]
114114
}
115115

116116
provisioner "powershell" {
117-
environment_vars = ["CUDA_VERSION=13.0"]
117+
environment_vars = ["CUDA_VERSION=13.2"]
118118
scripts = [
119119
"${path.root}/scripts/Installers/Install-CUDA-Tools.ps1",
120120
]
121121
}
122122

123123
provisioner "powershell" {
124-
environment_vars = ["CUDA_VERSION=13.2"]
124+
environment_vars = ["CUDA_VERSION=13.4"]
125125
scripts = [
126126
"${path.root}/scripts/Installers/Install-CUDA-Tools.ps1",
127127
]

0 commit comments

Comments
 (0)