Update win cudnn to 9.7.1.26 for CUDA 12.6 as well (#6290)

tinglvv · web-flow · commit d0db06958a34 · 2025-02-13T12:23:52.000-05:00
follow up for pytorch/pytorch#146957 use 9.7.1.26 for CUDA 12.6 too to test the windows AMI cc @atalman
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" {

Original file line number	Diff line number	Diff line change
`@@ -31,7 +31,7 @@ Switch ($cudaVersion) {`
`31`	`31`	`}`
`32`	`32`	`"12.6" {`
`33`	`33`	`$toolkitInstaller = "cuda_12.6.2_560.94_windows.exe"`
`34`		`- $cudnn_subfolder = "cudnn-windows-x86_64-9.5.1.17_cuda12-archive"`
	`34`	`+ $cudnn_subfolder = "cudnn-windows-x86_64-9.7.1.26_cuda12-archive"`
`35`	`35`	`$installerArgs += " cuda_profiler_api_$cudaVersion nvjitlink_$cudaVersion"`
`36`	`36`	`}`
`37`	`37`	`"12.8" {`