diff --git a/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1 b/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1 index 79f9b2ada7..c95bbc290f 100644 --- a/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1 +++ b/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1 @@ -28,7 +28,7 @@ Switch ($cudaVersion) { } "11.8" { $toolkitInstaller = "cuda_11.8.0_522.06_windows.exe" - $cudnn_subfolder = "cudnn-windows-x86_64-8.5.0.96_cuda11-archive" + $cudnn_subfolder = "cudnn-windows-x86_64-8.7.0.84_cuda11-archive" $installerArgs += " cuda_profiler_api_$cudaVersion" } }