[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
diff --git a/WORKSPACE b/WORKSPACE
index edd67ae..412ba2f 100644
--- a/WORKSPACE
+++ b/WORKSPACE
@@ -243,7 +243,7 @@
new_local_repository(
name = "cudnn",
build_file = "@//third_party:cudnn.BUILD",
- path = "/usr/",
+ path = "/usr/local/cuda",
)
local_repository(
