mirror of
https://github.com/THU-MIG/yolov10.git
synced 2025-05-23 05:24:22 +08:00
f6e393c1d2
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Noobtoss <96134731+Noobtoss@users.noreply.github.com> Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
29 lines
547 B
CMake
29 lines
547 B
CMake
cmake_minimum_required(VERSION 3.5)
|
|
|
|
project(Yolov8CPPInference VERSION 0.1)
|
|
|
|
set(CMAKE_INCLUDE_CURRENT_DIR ON)
|
|
|
|
# CUDA
|
|
set(CUDA_TOOLKIT_ROOT_DIR "/usr/local/cuda")
|
|
find_package(CUDA 11 REQUIRED)
|
|
|
|
set(CMAKE_CUDA_STANDARD 11)
|
|
set(CMAKE_CUDA_STANDARD_REQUIRED ON)
|
|
# !CUDA
|
|
|
|
# OpenCV
|
|
find_package(OpenCV REQUIRED)
|
|
include_directories(${OpenCV_INCLUDE_DIRS})
|
|
# !OpenCV
|
|
|
|
set(PROJECT_SOURCES
|
|
main.cpp
|
|
|
|
inference.h
|
|
inference.cpp
|
|
)
|
|
|
|
add_executable(Yolov8CPPInference ${PROJECT_SOURCES})
|
|
target_link_libraries(Yolov8CPPInference ${OpenCV_LIBS})
|