CMakeLists.txt 547 B

12345678910111213141516171819202122232425262728
  1. cmake_minimum_required(VERSION 3.5)
  2. project(Yolov8CPPInference VERSION 0.1)
  3. set(CMAKE_INCLUDE_CURRENT_DIR ON)
  4. # CUDA
  5. set(CUDA_TOOLKIT_ROOT_DIR "/usr/local/cuda")
  6. find_package(CUDA 11 REQUIRED)
  7. set(CMAKE_CUDA_STANDARD 11)
  8. set(CMAKE_CUDA_STANDARD_REQUIRED ON)
  9. # !CUDA
  10. # OpenCV
  11. find_package(OpenCV REQUIRED)
  12. include_directories(${OpenCV_INCLUDE_DIRS})
  13. # !OpenCV
  14. set(PROJECT_SOURCES
  15. main.cpp
  16. inference.h
  17. inference.cpp
  18. )
  19. add_executable(Yolov8CPPInference ${PROJECT_SOURCES})
  20. target_link_libraries(Yolov8CPPInference ${OpenCV_LIBS})