diff options
author | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-13 13:28:55 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-13 13:28:55 -0700 |
commit | 81494bc9081f871434b5b2fb6e04bc45a330972f (patch) | |
tree | 165cccdd313a6866c48b8d8957f33dbb88d289b8 /third_party | |
parent | 35535e6313c7c35a851466efd67be0ec1df14c9e (diff) | |
parent | b7d97e85a509dc14f3e81bb289fdb7bad64f0bbe (diff) |
Merge pull request #21499 from nmoran:debian-cuda-config
PiperOrigin-RevId: 208532594
Diffstat (limited to 'third_party')
-rw-r--r-- | third_party/gpus/cuda_configure.bzl | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/third_party/gpus/cuda_configure.bzl b/third_party/gpus/cuda_configure.bzl index e848fa175c..f6a39aeaf1 100644 --- a/third_party/gpus/cuda_configure.bzl +++ b/third_party/gpus/cuda_configure.bzl @@ -61,6 +61,7 @@ CUDA_LIB_PATHS = [ CUPTI_HEADER_PATHS = [ "extras/CUPTI/include/", "include/cuda/CUPTI/", + "include/", ] # Lookup paths for the cupti library, relative to the @@ -69,7 +70,7 @@ CUPTI_HEADER_PATHS = [ # the other CUDA libraries but rather in a special extras/CUPTI directory. CUPTI_LIB_PATHS = [ "extras/CUPTI/lib64/", - "lib/x86_64-linux-gnu", + "lib/x86_64-linux-gnu/", "lib64/", "extras/CUPTI/libx64/", "extras/CUPTI/lib/", @@ -96,6 +97,7 @@ CUDNN_INCLUDE_PATHS = [ NVVM_LIBDEVICE_PATHS = [ "nvvm/libdevice/", "share/cuda/", + "lib/nvidia-cuda-toolkit/libdevice/", ] # Files used to detect the NVVM libdevice path. |