mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
4ea7430ffb
- 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
27 lines
578 B
Plaintext
27 lines
578 B
Plaintext
# Adopted from: https://github.com/NVIDIA/TRTorch/blob/master/third_party/cudnn/local/BUILD
|
|
|
|
cc_library(
|
|
name = "cudnn_headers",
|
|
hdrs = ["include/cudnn.h"] + glob([
|
|
"include/cudnn+.h",
|
|
"include/cudnn_*.h",
|
|
]),
|
|
includes = ["include/"],
|
|
visibility = ["//visibility:private"],
|
|
)
|
|
|
|
cc_import(
|
|
name = "cudnn_lib",
|
|
shared_library = "lib64/libcudnn.so",
|
|
visibility = ["//visibility:private"],
|
|
)
|
|
|
|
cc_library(
|
|
name = "cudnn",
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
"cudnn_headers",
|
|
"cudnn_lib",
|
|
],
|
|
)
|