mirror of
https://github.com/opencv/opencv.git
synced 2024-11-29 05:29:54 +08:00
ae279966c2
add cuDNN dependency and setup build for cuda4dnn (#14660) * update cmake for cuda4dnn - Adds FindCUDNN - Adds new options: * WITH_CUDA * OPENCV_DNN_CUDA - Adds CUDA4DNN preprocessor symbol for the DNN module * FIX: append EXCLUDE_CUDA instead of overwrite * remove cuDNN dependency for user apps * fix unused variable warning
57 lines
1.8 KiB
CMake
57 lines
1.8 KiB
CMake
# Version Compute Capability from which OpenCV has been compiled is remembered
|
|
set(OpenCV_COMPUTE_CAPABILITIES "@OpenCV_CUDA_CC@")
|
|
|
|
set(OpenCV_CUDA_VERSION "@CUDA_VERSION_STRING@")
|
|
set(OpenCV_USE_CUBLAS "@HAVE_CUBLAS@")
|
|
set(OpenCV_USE_CUFFT "@HAVE_CUFFT@")
|
|
set(OpenCV_USE_NVCUVID "@HAVE_NVCUVID@")
|
|
|
|
set(OpenCV_CUDNN_VERSION "@CUDNN_VERSION@")
|
|
set(OpenCV_USE_CUDNN "@HAVE_CUDNN@")
|
|
|
|
if(NOT CUDA_FOUND)
|
|
find_host_package(CUDA ${OpenCV_CUDA_VERSION} EXACT REQUIRED)
|
|
else()
|
|
if(NOT CUDA_VERSION_STRING VERSION_EQUAL OpenCV_CUDA_VERSION)
|
|
message(FATAL_ERROR "OpenCV static library was compiled with CUDA ${OpenCV_CUDA_VERSION} support. Please, use the same version or rebuild OpenCV with CUDA ${CUDA_VERSION_STRING}")
|
|
endif()
|
|
endif()
|
|
|
|
set(OpenCV_CUDA_LIBS_ABSPATH ${CUDA_LIBRARIES})
|
|
|
|
if(CUDA_VERSION VERSION_LESS "5.5")
|
|
list(APPEND OpenCV_CUDA_LIBS_ABSPATH ${CUDA_npp_LIBRARY})
|
|
else()
|
|
find_cuda_helper_libs(nppc)
|
|
find_cuda_helper_libs(nppi)
|
|
find_cuda_helper_libs(npps)
|
|
list(APPEND OpenCV_CUDA_LIBS_ABSPATH ${CUDA_nppc_LIBRARY} ${CUDA_nppi_LIBRARY} ${CUDA_npps_LIBRARY})
|
|
endif()
|
|
|
|
if(OpenCV_USE_CUBLAS)
|
|
list(APPEND OpenCV_CUDA_LIBS_ABSPATH ${CUDA_CUBLAS_LIBRARIES})
|
|
endif()
|
|
|
|
if(OpenCV_USE_CUFFT)
|
|
list(APPEND OpenCV_CUDA_LIBS_ABSPATH ${CUDA_CUFFT_LIBRARIES})
|
|
endif()
|
|
|
|
if(OpenCV_USE_NVCUVID)
|
|
list(APPEND OpenCV_CUDA_LIBS_ABSPATH ${CUDA_nvcuvid_LIBRARIES})
|
|
endif()
|
|
|
|
if(WIN32)
|
|
list(APPEND OpenCV_CUDA_LIBS_ABSPATH ${CUDA_nvcuvenc_LIBRARIES})
|
|
endif()
|
|
|
|
set(OpenCV_CUDA_LIBS_RELPATH "")
|
|
foreach(l ${OpenCV_CUDA_LIBS_ABSPATH})
|
|
get_filename_component(_tmp ${l} PATH)
|
|
if(NOT ${_tmp} MATCHES "-Wl.*")
|
|
list(APPEND OpenCV_CUDA_LIBS_RELPATH ${_tmp})
|
|
endif()
|
|
endforeach()
|
|
|
|
list(REMOVE_DUPLICATES OpenCV_CUDA_LIBS_RELPATH)
|
|
link_directories(${OpenCV_CUDA_LIBS_RELPATH})
|