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

Add /usr/local/cudnn to the directories to look for CUDNN headers/libs.

Browse Source 
PiperOrigin-RevId: 279157906
Change-Id: I78035bed8ad06564424f142c984e784080ad8c00
This commit is contained in:
Christian Sigg 2019-11-07 13:42:03 -08:00 committed by TensorFlower Gardener
parent 8f00812c22
commit 37cdb806bc
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
third_party/gpus/find_cuda_config.py vendored
View File

@ -163,8 +163,8 @@ def _get_default_cuda_paths(cuda_version):
"C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v%s\\" %
cuda_version)
]
return ["/usr/local/cuda-%s" % cuda_version, "/usr/local/cuda", "/usr"
] + _get_ld_config_paths()
return ["/usr/local/cuda-%s" % cuda_version, "/usr/local/cuda", "/usr",
"/usr/local/cudnn"] + _get_ld_config_paths()
def _header_paths():