diff --git a/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1 b/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1 index 81d94a20b6..12d570d4e0 100644 --- a/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1 +++ b/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1 @@ -35,7 +35,7 @@ Switch ($cudaVersion) { $installerArgs += " cuda_profiler_api_$cudaVersion nvjitlink_$cudaVersion" } "12.8" { - $toolkitInstaller = "cuda_12.8.0_571.96_windows.exe" + $toolkitInstaller = "cuda_12.8.1_572.61_windows.exe" $cudnn_subfolder = "cudnn-windows-x86_64-9.7.1.26_cuda12-archive" $installerArgs += " cuda_profiler_api_$cudaVersion nvjitlink_$cudaVersion" }