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

improve CMake alternative, add instructions in README

Browse Source
This commit is contained in:
Stefano Sinigardi
2019-02-17 23:56:25 +01:00
parent c88ae98289
commit cbc1c6d8ca
6 changed files with 179 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
cmake/Modules/FindCUDNN.cmake
View File

@ -30,13 +30,17 @@ include(FindPackageHandleStandardArgs)
#set(CUDNN_ROOT_DIR "" CACHE PATH "Folder contains NVIDIA cuDNN")
find_path(CUDNN_INCLUDE_DIR cudnn.h
if(NOT CUDNN_INCLUDE_DIR)
find_path(CUDNN_INCLUDE_DIR cudnn.h
HINTS ${CUDNN_ROOT_DIR} ${CUDA_HOME} ${CUDA_TOOLKIT_ROOT_DIR}
PATH_SUFFIXES cuda/include include)
endif()
find_library(CUDNN_LIBRARY cudnn
if(NOT CUDNN_LIBRARY)
find_library(CUDNN_LIBRARY cudnn
HINTS ${CUDNN_ROOT_DIR} ${CUDA_HOME} ${CUDA_TOOLKIT_ROOT_DIR}
PATH_SUFFIXES lib lib64 cuda/lib cuda/lib64 lib/x64)
endif()
find_package_handle_standard_args(
CUDNN DEFAULT_MSG CUDNN_INCLUDE_DIR CUDNN_LIBRARY)