CMakeLists.txt 707 B

1234567891011121314151617
  1. cmake_minimum_required(VERSION 3.10)
  2. set(CMAKE_CXX_STANDARD 14)
  3. set(CMAKE_CXX_STANDARD_REQUIRED TRUE)
  4. set(CMAKE_MODULE_PATH "${CMAKE_SOURCE_DIR}" ${CMAKE_MODULE_PATH})
  5. project(TensorRTSample)
  6. find_package(Threads)
  7. find_package(CUDA REQUIRED)
  8. find_package(OpenCV REQUIRED)
  9. find_package(TensorRT REQUIRED)
  10. # TARGETS --------------------------------------------------------------------------------------------------------------
  11. add_executable(trt_sample trt_sample.cpp)
  12. target_include_directories(trt_sample PUBLIC ${OpenCV_INCLUDE_DIRS} ${CUDA_INCLUDE_DIRS} ${TensorRT_INCLUDE_DIRS})
  13. target_link_libraries(trt_sample PUBLIC ${OpenCV_LIBS} ${CUDA_LIBRARIES} ${CMAKE_THREAD_LIBS_INIT} ${TensorRT_LIBRARIES})