diff options
Diffstat (limited to 'third_party/gpus/cuda/cuda_config.sh')
-rwxr-xr-x | third_party/gpus/cuda/cuda_config.sh | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/third_party/gpus/cuda/cuda_config.sh b/third_party/gpus/cuda/cuda_config.sh index 21f36d4416..42cd254644 100755 --- a/third_party/gpus/cuda/cuda_config.sh +++ b/third_party/gpus/cuda/cuda_config.sh @@ -133,8 +133,10 @@ if test -e ${CUDNN_INSTALL_PATH}/cudnn.h; then CUDNN_HEADER_PATH=${CUDNN_INSTALL_PATH} elif test -e ${CUDNN_INSTALL_PATH}/include/cudnn.h; then CUDNN_HEADER_PATH=${CUDNN_INSTALL_PATH}/include +elif test -e /usr/include/cudnn.h; then + CUDNN_HEADER_PATH=/usr/include else - CudnnError "cannot find cudnn.h under: ${CUDNN_INSTALL_PATH}" + CudnnError "cannot find cudnn.h under: ${CUDNN_INSTALL_PATH} or /usr/include" fi # Locate libcudnn.so.${$TF_CUDNN_VERSION} |