[BE] Don't copy CuDNN libs twice (#115872)

- It was installed twice : once in `/usr/local/cuda/lib64` folder and 2nd time in `/usr/lib64`
- And don't install CuDNN headers thrice, only in `/usr/local/cuda/includa`
- Error on unknown CUDA version
- Modify bazel builds to look for cudnn in `/usr/local/cuda` folder
Pull Request resolved: https://github.com/pytorch/pytorch/pull/115872
Approved by: https://github.com/huydhn
This commit is contained in:
Nikita Shulga
2023-12-15 09:47:10 +00:00
committed by PyTorch MergeBot
parent b4d6443bcf
commit 4ea7430ffb
3 changed files with 7 additions and 10 deletions

View File

@ -243,7 +243,7 @@ new_local_repository(
new_local_repository(
name = "cudnn",
build_file = "@//third_party:cudnn.BUILD",
path = "/usr/",
path = "/usr/local/cuda",
)
local_repository(