OpenCVConfig-CUDALanguage.cmake.in 1.3 KB

12345678910111213141516171819202122232425262728293031
  1. # Version Compute Capability from which OpenCV has been compiled is remembered
  2. set(OpenCV_COMPUTE_CAPABILITIES "@OpenCV_CUDA_CC@")
  3. set(OpenCV_CUDA_VERSION "@CUDA_VERSION_STRING@")
  4. set(OpenCV_USE_CUBLAS "@HAVE_CUBLAS@")
  5. set(OpenCV_USE_CUFFT "@HAVE_CUFFT@")
  6. set(OpenCV_USE_NVCUVID "@HAVE_NVCUVID@")
  7. set(OpenCV_USE_NVCUVENC "@HAVE_NVCUVENC@")
  8. set(OpenCV_CUDNN_VERSION "@CUDNN_VERSION@")
  9. set(OpenCV_USE_CUDNN "@HAVE_CUDNN@")
  10. set(ENABLE_CUDA_FIRST_CLASS_LANGUAGE ON)
  11. if(NOT CUDAToolkit_FOUND)
  12. if(NOT CMAKE_VERSION VERSION_LESS 3.18)
  13. if(UNIX AND NOT CMAKE_CUDA_COMPILER AND NOT CUDAToolkit_ROOT)
  14. message(STATUS "Checking for CUDAToolkit in default location (/usr/local/cuda)")
  15. set(CUDA_PATH "/usr/local/cuda" CACHE INTERNAL "")
  16. set(ENV{CUDA_PATH} ${CUDA_PATH})
  17. endif()
  18. find_package(CUDAToolkit ${OpenCV_CUDA_VERSION} EXACT REQUIRED)
  19. else()
  20. message(FATAL_ERROR "Using OpenCV compiled with CUDA as first class language requires CMake \>= 3.18.")
  21. endif()
  22. else()
  23. if(CUDAToolkit_FOUND)
  24. set(CUDA_VERSION_STRING ${CUDAToolkit_VERSION})
  25. endif()
  26. if(NOT CUDA_VERSION_STRING VERSION_EQUAL OpenCV_CUDA_VERSION)
  27. 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}")
  28. endif()
  29. endif()