diff --git a/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1 b/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1 index 4e17113000..81d94a20b6 100644 --- a/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1 +++ b/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1 @@ -31,7 +31,7 @@ Switch ($cudaVersion) { } "12.6" { $toolkitInstaller = "cuda_12.6.2_560.94_windows.exe" - $cudnn_subfolder = "cudnn-windows-x86_64-9.5.1.17_cuda12-archive" + $cudnn_subfolder = "cudnn-windows-x86_64-9.7.1.26_cuda12-archive" $installerArgs += " cuda_profiler_api_$cudaVersion nvjitlink_$cudaVersion" } "12.8" {