Commit 766c46b5 authored by elda27's avatar elda27 Committed by Davis E. King

find_cudnn modified for cudnn installed a user directory. (#275)

parent 9635f785
......@@ -11,6 +11,6 @@ get_filename_component(cudnn_hint_path "${CUDA_CUBLAS_LIBRARIES}" PATH)
find_library(cudnn cudnn
HINTS ${cudnn_hint_path} ENV CUDNN_LIBRARY_DIR ENV CUDNN_HOME
PATHS /usr/local /usr/local/cuda ENV LD_LIBRARY_PATH
PATH_SUFFIXES lib64 lib
PATH_SUFFIXES lib64 lib x64
)
mark_as_advanced(cudnn cudnn_include)
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment