diff --git a/CMakeLists.txt b/CMakeLists.txt index 97134f4760..c30845e4f1 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -456,6 +456,18 @@ if(WITH_OPENCL) include(cmake/OpenCVDetectOpenCL.cmake) endif() +# ---------------------------------------------------------------------------- +# Add CUDA libraries (needed for apps/tools, samples) +# ---------------------------------------------------------------------------- +if(HAVE_CUDA) + set(OPENCV_LINKER_LIBS ${OPENCV_LINKER_LIBS} ${CUDA_LIBRARIES} ${CUDA_npp_LIBRARY}) + if(HAVE_CUBLAS) + set(OPENCV_LINKER_LIBS ${OPENCV_LINKER_LIBS} ${CUDA_cublas_LIBRARY}) + endif() + if(HAVE_CUFFT) + set(OPENCV_LINKER_LIBS ${OPENCV_LINKER_LIBS} ${CUDA_cufft_LIBRARY}) + endif() +endif() # ---------------------------------------------------------------------------- # Solution folders: # ---------------------------------------------------------------------------- diff --git a/apps/CMakeLists.txt b/apps/CMakeLists.txt index 217490b2de..347dbac7f5 100644 --- a/apps/CMakeLists.txt +++ b/apps/CMakeLists.txt @@ -1,2 +1,4 @@ +link_libraries(${OPENCV_LINKER_LIBS}) + add_subdirectory(haartraining) add_subdirectory(traincascade)