update 12.6 win cudnn to 9.7.1.26

pytorch · Feb 13, 2025 · f9f7bde · f9f7bde
1 parent 0c4b50f
commit f9f7bde
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
@@ -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" {