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.
46 lines
1.1 KiB
46 lines
1.1 KiB
cmake_minimum_required (VERSION 3.5)
|
|
find_package( OpenCV REQUIRED )
|
|
|
|
message(STATUS "oPENCV Library status:")
|
|
message(STATUS ">version:${OpenCV_VERSION}")
|
|
message(STATUS "Include:${OpenCV_INCLUDE_DIRS}")
|
|
|
|
|
|
set(CMAKE_CXX_STANDARD 17)
|
|
|
|
set(CMAKE_CXX_FLAGS "-Wno-error=deprecated-declarations -Wno-deprecated-declarations")
|
|
|
|
find_package(CUDA REQUIRED)
|
|
message(STATUS "cuda version: " ${CUDA_VERSION_STRING})
|
|
message(STATUS "cuda CUDA_INCLUDE_DIRS: " ${CUDA_INCLUDE_DIRS})
|
|
include_directories(${CUDA_INCLUDE_DIRS})
|
|
|
|
include_directories(
|
|
${CMAKE_CURRENT_SOURCE_DIR}/include
|
|
${PROJECT_SOURCE_DIR}/AIEngineModule/include
|
|
${PROJECT_SOURCE_DIR}/AIEngineModule/include_base
|
|
)
|
|
|
|
link_directories(
|
|
/usr/local/lib/
|
|
)
|
|
set(CMAKE_CUDA_ARCHITECTURES 86)
|
|
set(CMAKE_CUDA_COMPILER "/usr/local/cuda/bin/nvcc")
|
|
enable_language(CUDA)
|
|
file(GLOB SRC_LISTS
|
|
${CMAKE_CURRENT_SOURCE_DIR}/*.cpp
|
|
${PROJECT_SOURCE_DIR}/AIEngineModule/src/*.cpp
|
|
${PROJECT_SOURCE_DIR}/AIEngineModule/src/*.c
|
|
${PROJECT_SOURCE_DIR}/AIEngineModule/src/*.cu
|
|
)
|
|
|
|
add_executable("ai_test" ${SRC_LISTS})
|
|
|
|
target_link_libraries("ai_test"
|
|
pthread
|
|
z
|
|
nvinfer
|
|
${CUDA_LIBRARIES}
|
|
${OpenCV_LIBS}
|
|
)
|