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

automatically detect CUDA and use it, without any possible manual intervention

Browse Source
This commit is contained in:
Stefano Sinigardi
2019-02-20 11:42:53 +01:00
parent 9c64f885a8
commit f950085a45
5 changed files with 47 additions and 137 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
cmake/Modules/FindCUDNN.cmake
View File

@ -19,26 +19,18 @@
# ``CUDNN_LIBRARIES``
# The CUDNN libraries.
#
# Hints
# ^^^^^
#
# ``CUDNN_ROOT_DIR``
# Set this variable to a directory that contains a CUDNN installation.
#
include(FindPackageHandleStandardArgs)
#set(CUDNN_ROOT_DIR "" CACHE PATH "Folder contains NVIDIA cuDNN")
if(NOT CUDNN_INCLUDE_DIR)
find_path(CUDNN_INCLUDE_DIR cudnn.h
HINTS ${CUDNN_ROOT_DIR} ${CUDA_HOME} ${CUDA_TOOLKIT_ROOT_DIR}
HINTS ${CUDA_HOME} ${CUDA_TOOLKIT_ROOT_DIR}
PATH_SUFFIXES cuda/include include)
endif()
if(NOT CUDNN_LIBRARY)
find_library(CUDNN_LIBRARY cudnn
HINTS ${CUDNN_ROOT_DIR} ${CUDA_HOME} ${CUDA_TOOLKIT_ROOT_DIR}
HINTS ${CUDA_HOME} ${CUDA_TOOLKIT_ROOT_DIR}
PATH_SUFFIXES lib lib64 cuda/lib cuda/lib64 lib/x64)
endif()