Spaces:
Running
Running
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}) | |