Commit a9d4b15b authored by Daniel Crispell's avatar Daniel Crispell

Check CUDNN environment variables for path

parent 6f46af23
...@@ -3,12 +3,12 @@ message(STATUS "Looking for cuDNN install...") ...@@ -3,12 +3,12 @@ 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} HINTS ${CUDA_INCLUDE_DIRS} $ENV{CUDNN_INCLUDE_DIR}
PATHS /usr/local/include PATHS /usr/local/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} HINTS ${cudnn_hint_path} $ENV{CUDNN_LIBRARY_DIR}
PATHS /usr/local/lib64 PATHS /usr/local/lib64
/usr/local/cuda/lib64 /usr/local/cuda/lib64
/usr/local/cuda/lib /usr/local/cuda/lib
......
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