experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add more alternative path to find cuda library

Browse Source
This commit is contained in:
levinxo 2020-08-26 17:25:47 +08:00
parent 5ef2f641c7
commit 897aa38570
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
third_party/gpus/find_cuda_config.py vendored
View File

@ -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")