mirror of https://github.com/opencv/opencv.git
Open Source Computer Vision Library
https://opencv.org/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
32 lines
1.3 KiB
32 lines
1.3 KiB
2 years ago
|
# 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_USE_NVCUVENC "@HAVE_NVCUVENC@")
|
||
|
|
set(OpenCV_CUDNN_VERSION "@CUDNN_VERSION@")
|
||
|
|
set(OpenCV_USE_CUDNN "@HAVE_CUDNN@")
|
||
|
|
set(ENABLE_CUDA_FIRST_CLASS_LANGUAGE ON)
|
||
|
|
|
||
|
|
if(NOT CUDAToolkit_FOUND)
|
||
|
|
if(NOT CMAKE_VERSION VERSION_LESS 3.18)
|
||
|
|
if(UNIX AND NOT CMAKE_CUDA_COMPILER AND NOT CUDAToolkit_ROOT)
|
||
|
|
message(STATUS "Checking for CUDAToolkit in default location (/usr/local/cuda)")
|
||
|
|
set(CUDA_PATH "/usr/local/cuda" CACHE INTERNAL "")
|
||
|
|
set(ENV{CUDA_PATH} ${CUDA_PATH})
|
||
|
|
endif()
|
||
|
|
find_package(CUDAToolkit ${OpenCV_CUDA_VERSION} EXACT REQUIRED)
|
||
|
|
else()
|
||
|
|
message(FATAL_ERROR "Using OpenCV compiled with CUDA as first class language requires CMake \>= 3.18.")
|
||
|
|
endif()
|
||
|
|
else()
|
||
|
|
if(CUDAToolkit_FOUND)
|
||
|
|
set(CUDA_VERSION_STRING ${CUDAToolkit_VERSION})
|
||
|
|
endif()
|
||
|
|
if(NOT CUDA_VERSION_STRING VERSION_EQUAL OpenCV_CUDA_VERSION)
|
||
|
|
message(FATAL_ERROR "OpenCV library was compiled with CUDA ${OpenCV_CUDA_VERSION} support. Please, use the same version or rebuild OpenCV with CUDA ${CUDA_VERSION_STRING}")
|
||
|
|
endif()
|
||
|
|
endif()
|