Skip to content

Commit 5237e15

Browse files
authored
Fix CUDA 13.0 Windows AMI building (#7035)
CUDA 13.0 looks like there is an issue passing individual packages to install. From: https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html ``` The installer can be executed in silent mode by executing the package with the -s flag. Additional parameters can be passed which will install specific subpackages instead of all packages. See the table below for a list of all the CUDA Toolkit subpackage names. ``` Lets just install all packages for CUDA 13.0 - easier to maintain.
1 parent d7b46d5 commit 5237e15

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

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

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ $windowsS3BaseUrl = "https://ossci-windows.s3.amazonaws.com"
1010
$ProgressPreference = 'SilentlyContinue'
1111

1212
# installerArgs
13-
$installerArgs = "nvcc_$cudaVersion cuobjdump_$cudaVersion nvprune_$cudaVersion nvprof_$cudaVersion cupti_$cudaVersion cublas_$cudaVersion cublas_dev_$cudaVersion cudart_$cudaVersion cufft_$cudaVersion cufft_dev_$cudaVersion curand_$cudaVersion curand_dev_$cudaVersion cusolver_$cudaVersion cusolver_dev_$cudaVersion cusparse_$cudaVersion cusparse_dev_$cudaVersion npp_$cudaVersion npp_dev_$cudaVersion nvrtc_$cudaVersion nvrtc_dev_$cudaVersion nvml_dev_$cudaVersion nvjpeg_$cudaVersion nvjpeg_dev_$cudaVersion"
13+
$installerArgs = "nvcc_$cudaVersion cuobjdump_$cudaVersion nvprune_$cudaVersion nvprof_$cudaVersion cupti_$cudaVersion cublas_$cudaVersion cublas_dev_$cudaVersion cudart_$cudaVersion cufft_$cudaVersion cufft_dev_$cudaVersion curand_$cudaVersion curand_dev_$cudaVersion cusolver_$cudaVersion cusolver_dev_$cudaVersion cusparse_$cudaVersion cusparse_dev_$cudaVersion npp_$cudaVersion npp_dev_$cudaVersion nvrtc_$cudaVersion nvrtc_dev_$cudaVersion nvml_dev_$cudaVersion nvjpeg_$cudaVersion nvjpeg_dev_$cudaVersion cuda_profiler_api_$cudaVersion nvjitlink_$cudaVersion thrust_$cudaVersion"
1414

1515
# Switch statement for specfic CUDA versions
1616
$cudnn_subfolder="cuda"
@@ -21,26 +21,22 @@ $toolkitInstaller = "cuda_12.6.3_561.17_windows.exe"
2121
Switch ($cudaVersion) {
2222
"12.6" {
2323
$toolkitInstaller = "cuda_12.6.3_561.17_windows.exe"
24-
$installerArgs += " cuda_profiler_api_$cudaVersion nvjitlink_$cudaVersion"
2524
}
2625
"12.8" {
2726
$toolkitInstaller = "cuda_12.8.1_572.61_windows.exe"
28-
$installerArgs += " cuda_profiler_api_$cudaVersion nvjitlink_$cudaVersion"
2927
}
3028
"12.9" {
3129
$toolkitInstaller = "cuda_12.9.1_576.57_windows.exe"
32-
$installerArgs += " cuda_profiler_api_$cudaVersion nvjitlink_$cudaVersion"
3330
}
3431
"13.0" {
3532
$cudnn_subfolder="cudnn-windows-x86_64-9.12.0.46_cuda13-archive"
3633
$toolkitInstaller = "cuda_13.0.0_windows.exe"
37-
$installerArgs += " cuda_profiler_api_$cudaVersion nvjitlink_$cudaVersion"
34+
$installerArgs = ""
3835
}
3936
}
4037

4138

4239
$cudnnZip = "$cudnn_subfolder.zip"
43-
$installerArgs = "$installerArgs thrust_$cudaVersion"
4440
$cudnn_lib_folder="lib"
4541

4642
Write-Output "Downloading ZLIB DLL, $windowsS3BaseUrl/zlib123dllx64.zip"

0 commit comments

Comments
 (0)