diff --git a/cuDNN.cmake b/cuDNN.cmake index 49836b38..5570c28f 100644 --- a/cuDNN.cmake +++ b/cuDNN.cmake @@ -39,6 +39,7 @@ message(STATUS "Configuring cuDNN ...") find_path( _CUDNN_INCLUDE_DIR cudnn.h PATHS + $ENV{CUDNN_PATH} ${CUDA_TOOLKIT_ROOT_DIR}/include $ENV{CUDNN_PATH}/include $ENV{CUDA_PATH}/include @@ -48,6 +49,7 @@ find_path( find_library( _CUDNN_LIBRARY cudnn HINTS + $ENV{CUDNN_PATH} ${CUDA_TOOLKIT_ROOT_DIR}/lib64 ${CUDA_TOOLKIT_ROOT_DIR}/lib/x64 ${CUDA_TOOLKIT_ROOT_DIR}/lib