Skip to content

Commit

Permalink
Fix cudnn 11.8 lib path on windows AMI (#1914)
Browse files Browse the repository at this point in the history
For cudnn-windows-x86_64-8.7.0.84_cuda11-archive libs are located under
lib/x64
For cudnn-windows-x86_64-8.5.0.96_cuda11-archive libs are located under
lib/

Failing test (without this fix):

https://github.com/pytorch/pytorch-canary/actions/runs/4027562763/jobs/6925900401

Successful test (with this fix):

https://github.com/pytorch/pytorch-canary/actions/runs/4027562763/jobs/6954158425

Test PR: 
pytorch/pytorch-canary#161

AMI Update PR: 
pytorch-labs/pytorch-gha-infra#107
  • Loading branch information
atalman authored Jan 30, 2023
1 parent 7ca1086 commit 61d67bf
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion aws/ami/windows/scripts/Installers/Install-CUDA-Tools.ps1
Original file line number Diff line number Diff line change
Expand Up @@ -96,7 +96,11 @@ function Install-Cudnn() {
Write-Output "Copying cudnn to $expectedInstallLocation"

Copy-Item -Force -Verbose -Recurse "$tmpCudnnExtracted\$cudnn_subfolder\bin\*" "$expectedInstallLocation\bin"
Copy-Item -Force -Verbose -Recurse "$tmpCudnnExtracted\$cudnn_subfolder\$cudnn_lib_folder\*" "$expectedInstallLocation\lib\x64"
if ($cudaVersion -eq "11.8") {
Copy-Item -Force -Verbose -Recurse "$tmpCudnnExtracted\$cudnn_subfolder\$cudnn_lib_folder\x64\*" "$expectedInstallLocation\lib\x64"
} else {
Copy-Item -Force -Verbose -Recurse "$tmpCudnnExtracted\$cudnn_subfolder\$cudnn_lib_folder\*" "$expectedInstallLocation\lib\x64"
}
Copy-Item -Force -Verbose -Recurse "$tmpCudnnExtracted\$cudnn_subfolder\include\*" "$expectedInstallLocation\include"

if (-Not (Test-Path -Path "$expectedInstallLocation\include\cudnn.h" -PathType Leaf)) {
Expand Down

0 comments on commit 61d67bf

Please sign in to comment.