Windows AMI update cudnn-windows-x86_64-9.5.1.17_cuda12-archive (#5924)

Cuda 12.6 migration
pytorch · Nov 15, 2024 · 32a72fa · 32a72fa
1 parent 9abd4d9
commit 32a72fa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1 b/aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1
@@ -36,7 +36,7 @@ Switch ($cudaVersion) {
   }
   "12.6" {
     $toolkitInstaller = "cuda_12.6.2_560.94_windows.exe"
-    $cudnn_subfolder = "cudnn-windows-x86_64-9.5.0.50_cuda12-archive"
+    $cudnn_subfolder = "cudnn-windows-x86_64-9.5.1.17_cuda12-archive"
     $installerArgs += " cuda_profiler_api_$cudaVersion nvjitlink_$cudaVersion"
   }
 }