Upgrade cudnn for cuda 11.8 (#1523)

Upgrade cudnn for cuda 11.8 Related to: pytorch/builder#1271
pytorch · Jan 26, 2023 · 4c6b3db · 4c6b3db
1 parent c5dd068
commit 4c6b3db
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
@@ -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"
   }
 }