Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@ $installerArgs = "nvcc_$cudaVersion cuobjdump_$cudaVersion nvprune_$cudaVersion
# Switch statement for specfic CUDA versions
$cudnn_subfolder="cuda"
$cudnn_lib_folder="lib\x64"
$cudnn_subfolder="cudnn-windows-x86_64-9.10.2.21_cuda12-archive"

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

As discussed please keep older cudnn for 12.6-12.9 builds

$cudnn_subfolder="cudnn-windows-x86_64-9.12.0.46_cuda12-archive"
$toolkitInstaller = "cuda_12.6.3_561.17_windows.exe"

Switch ($cudaVersion) {
Expand All @@ -32,6 +32,7 @@ Switch ($cudaVersion) {
$installerArgs += " cuda_profiler_api_$cudaVersion nvjitlink_$cudaVersion"
}
"13.0" {
$cudnn_subfolder="cudnn-windows-x86_64-9.12.0.46_cuda13-archive"
$toolkitInstaller = "cuda_13.0.0_windows.exe"
$installerArgs += " cuda_profiler_api_$cudaVersion nvjitlink_$cudaVersion"
}
Expand Down