mirror of
https://github.com/pjreddie/darknet.git
synced 2023-08-10 21:13:14 +03:00
[FindCUDNN] add another hint for newly suggested ENV variable
This commit is contained in:
@ -24,13 +24,13 @@ include(FindPackageHandleStandardArgs)
|
|||||||
|
|
||||||
if(NOT CUDNN_INCLUDE_DIR)
|
if(NOT CUDNN_INCLUDE_DIR)
|
||||||
find_path(CUDNN_INCLUDE_DIR cudnn.h
|
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)
|
PATH_SUFFIXES cuda/include include)
|
||||||
endif()
|
endif()
|
||||||
|
|
||||||
if(NOT CUDNN_LIBRARY)
|
if(NOT CUDNN_LIBRARY)
|
||||||
find_library(CUDNN_LIBRARY cudnn
|
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)
|
PATH_SUFFIXES lib lib64 cuda/lib cuda/lib64 lib/x64)
|
||||||
endif()
|
endif()
|
||||||
|
|
||||||
|
Reference in New Issue
Block a user