Commit c772e4ae authored by Davis King's avatar Davis King

Made CMake's search for cuDNN a little more broad

parent 0a80fdd1
...@@ -3,15 +3,14 @@ message(STATUS "Looking for cuDNN install...") ...@@ -3,15 +3,14 @@ message(STATUS "Looking for cuDNN install...")
# Look for cudnn, we will look in the same place as other CUDA # Look for cudnn, we will look in the same place as other CUDA
# libraries and also a few other places as well. # libraries and also a few other places as well.
find_path(cudnn_include cudnn.h find_path(cudnn_include cudnn.h
HINTS ${CUDA_INCLUDE_DIRS} $ENV{CUDNN_INCLUDE_DIR} HINTS ${CUDA_INCLUDE_DIRS} ENV CUDNN_INCLUDE_DIR ENV CUDNN_HOME
PATHS /usr/local/include PATHS /usr/local ENV CPATH
PATH_SUFFIXES include
) )
get_filename_component(cudnn_hint_path ${CUDA_CUBLAS_LIBRARIES} PATH) get_filename_component(cudnn_hint_path ${CUDA_CUBLAS_LIBRARIES} PATH)
find_library(cudnn cudnn find_library(cudnn cudnn
HINTS ${cudnn_hint_path} $ENV{CUDNN_LIBRARY_DIR} HINTS ${cudnn_hint_path} ENV CUDNN_LIBRARY_DIR ENV CUDNN_HOME
PATHS /usr/local/lib64 PATHS /usr/local /usr/local/cuda ENV LD_LIBRARY_PATH
/usr/local/cuda/lib64 PATH_SUFFIXES lib64 lib
/usr/local/cuda/lib
/usr/local/lib
) )
mark_as_advanced(cudnn cudnn_include) 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