CMakeLists.txt 282 B

1234567891011121314
  1. cmake_minimum_required(VERSION 3.1)
  2. project(trt_image_classification)
  3. find_package(OpenCV REQUIRED)
  4. find_package(CUDA REQUIRED)
  5. include_directories(${CMAKE_SOURCE_DIR})
  6. set(CUDA_NVCC_FLAGS --std=c++11)
  7. set(CMAKE_CXX_STANDARD 11)
  8. add_subdirectory(examples)
  9. add_subdirectory(src)