Skip to content

Commit 03a1d14

Browse files
authored
Add CUDA 12.9 Windows AMI (#6734)
pytorch/pytorch#155196 cc @atalman
1 parent 58b40fc commit 03a1d14

File tree

2 files changed

+8
-8
lines changed

2 files changed

+8
-8
lines changed

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

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -24,11 +24,6 @@ Switch ($cudaVersion) {
2424
$cudnn_subfolder = "cudnn-windows-x86_64-9.1.0.70_cuda11-archive"
2525
$installerArgs += " cuda_profiler_api_$cudaVersion"
2626
}
27-
"12.4" {
28-
$toolkitInstaller = "cuda_12.4.0_551.61_windows.exe"
29-
$cudnn_subfolder = "cudnn-windows-x86_64-9.1.0.70_cuda12-archive"
30-
$installerArgs += " cuda_profiler_api_$cudaVersion nvjitlink_$cudaVersion"
31-
}
3227
"12.6" {
3328
$toolkitInstaller = "cuda_12.6.2_560.94_windows.exe"
3429
$cudnn_subfolder = "cudnn-windows-x86_64-9.7.1.26_cuda12-archive"
@@ -39,6 +34,11 @@ Switch ($cudaVersion) {
3934
$cudnn_subfolder = "cudnn-windows-x86_64-9.7.1.26_cuda12-archive"
4035
$installerArgs += " cuda_profiler_api_$cudaVersion nvjitlink_$cudaVersion"
4136
}
37+
"12.9" {
38+
$toolkitInstaller = "cuda_12.9.1_576.57_windows.exe"
39+
$cudnn_subfolder = "cudnn-windows-x86_64-9.10.2.21_cuda12-archive"
40+
$installerArgs += " cuda_profiler_api_$cudaVersion nvjitlink_$cudaVersion"
41+
}
4242
}
4343

4444

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.4"]
110+
environment_vars = ["CUDA_VERSION=12.6"]
111111
scripts = [
112112
"${path.root}/scripts/Installers/Install-CUDA-Tools.ps1",
113113
]
114114
}
115115

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

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

0 commit comments

Comments
 (0)