Upgrade cudnn for cuda 11.8 (#1523)

atalman · web-flow · commit 4c6b3db4b4d9 · 2023-01-26T17:57:53.000-05:00
Upgrade cudnn for cuda 11.8 Related to: pytorch/builder#1271
diff --git 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"
   }
 }

Original file line number	Diff line number	Diff line change
`@@ -28,7 +28,7 @@ Switch ($cudaVersion) {`
`28`	`28`	`}`
`29`	`29`	`"11.8" {`
`30`	`30`	`$toolkitInstaller = "cuda_11.8.0_522.06_windows.exe"`
`31`		`- $cudnn_subfolder = "cudnn-windows-x86_64-8.5.0.96_cuda11-archive"`
	`31`	`+ $cudnn_subfolder = "cudnn-windows-x86_64-8.7.0.84_cuda11-archive"`
`32`	`32`	`$installerArgs += " cuda_profiler_api_$cudaVersion"`
`33`	`33`	`}`
`34`	`34`	`}`