From 897aa385703eb45afb6006ca701c6d51c2f8d835 Mon Sep 17 00:00:00 2001 From: levinxo Date: Wed, 26 Aug 2020 17:25:47 +0800 Subject: [PATCH] Add more alternative path to find cuda library --- third_party/gpus/find_cuda_config.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/third_party/gpus/find_cuda_config.py b/third_party/gpus/find_cuda_config.py index 091cd32d5fe..80f343023cd 100644 --- a/third_party/gpus/find_cuda_config.py +++ b/third_party/gpus/find_cuda_config.py @@ -176,6 +176,7 @@ def _header_paths(): "include/*-linux-gnu", "extras/CUPTI/include", "include/cuda/CUPTI", + "local/cuda/extras/CUPTI/include", ] @@ -188,6 +189,8 @@ def _library_paths(): "lib/*-linux-gnu", "lib/x64", "extras/CUPTI/*", + "local/cuda/lib64", + "local/cuda/extras/CUPTI/lib64", ] @@ -268,12 +271,14 @@ def _find_cuda_config(base_paths, required_version): nvcc_path, nvcc_version = _find_versioned_file(base_paths, [ "", "bin", + "local/cuda/bin", ], nvcc_name, cuda_version, get_nvcc_version) nvvm_path = _find_file(base_paths, [ "nvvm/libdevice", "share/cuda", "lib/nvidia-cuda-toolkit/libdevice", + "local/cuda/nvvm/libdevice", ], "libdevice*.10.bc") cupti_header_path = _find_file(base_paths, _header_paths(), "cupti.h")