mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

[FindCUDNN] add another hint for newly suggested ENV variable

Browse Source
This commit is contained in:
Stefano Sinigardi
2019-02-22 09:14:41 +01:00
parent 20757e217f
commit d8291858d5
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
cmake/Modules/FindCUDNN.cmake
View File

@ -24,13 +24,13 @@ include(FindPackageHandleStandardArgs)
if(NOT CUDNN_INCLUDE_DIR)
find_path(CUDNN_INCLUDE_DIR cudnn.h
HINTS ${CUDA_HOME} ${CUDA_TOOLKIT_ROOT_DIR} $ENV{cudnn}
HINTS ${CUDA_HOME} ${CUDA_TOOLKIT_ROOT_DIR} $ENV{cudnn} $ENV{CUDNN}
PATH_SUFFIXES cuda/include include)
endif()
if(NOT CUDNN_LIBRARY)
find_library(CUDNN_LIBRARY cudnn
HINTS ${CUDA_HOME} ${CUDA_TOOLKIT_ROOT_DIR} $ENV{cudnn}
HINTS ${CUDA_HOME} ${CUDA_TOOLKIT_ROOT_DIR} $ENV{cudnn} $ENV{CUDNN}
PATH_SUFFIXES lib lib64 cuda/lib cuda/lib64 lib/x64)
endif()