cmake_minimum_required(VERSION 3.5)
|
project(yolo_SDK)
|
set(EXECUTABLE_OUTPUT_PATH ${PROJECT_SOURCE_DIR}/./build)
|
set(CMAKE_CXX_STANDARD 11)
|
|
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -DGPU -DCUDNN -DOPENMP -march=native -W -pthread")
|
|
add_compile_options(-fPIC -g -Wall -Wshadow -Wno-sign-compare -w)
|
|
#set(TENSORFLOW_DIR /home/disk1/s_opt/tensorflow/tensorflow-1.14.0)
|
set(PYTORCH_DIR /home/disk1/s_opt/03_libtorch/libtorch)
|
set(OPENCV_DIR /home/disk1/s_opt/01_opencv/opencv2.4.13)
|
|
SET(SOURCES
|
|
)
|
|
#cudaS
|
find_package(CUDA REQUIRED)
|
|
message("-- CUDA version:$(CUDA_VERSION)")
|
|
# 设置环境的模型查找，如果CUDA版本大于9,则执行CUDA环境的模型查找
|
if(CUDA_VERSION_MAJOR GREATER 9)
|
message("-- CUDA ${CUDA_VERSION_MAJOR} detected")
|
set(
|
CUDA_NVCC_FLAGS
|
${CUDA_NVCC_FLAGS};
|
-gencode arch=compute_61,code=sm_61 # 不同GPU有不同的算力指数，可查看算力表
|
)
|
set(CUDA_NVCC_FLAGS_RELWITHDEBINFO "--device-debug;-lineinfo")
|
|
#find_package(OpenCV REQUIRED) # 查找系统的默认opencv环境
|
#message(${OpenCV_LIBS})
|
endif()
|
|
include_directories(/home/disk1/s_opt/04_protobuf/protobuf-3.10.0/01_protobuf/include)
|
link_directories(/home/disk1/s_opt/04_protobuf/protobuf-3.10.0/01_protobuf/lib) #动态链接库目录/bazel-bin/tensorflow
|
|
link_directories(
|
/usr/local/lib
|
/usr/local/cuda-10.0/lib64
|
# /usr/lib/x86_64-linux-gnu/
|
# ${CMAKE_SOURCE_DIR}/opencv3.4.8/lib
|
${OPENCV_DIR}/lib
|
./
|
)
|
|
SET(LIBS
|
yolo
|
protobuf
|
torch
|
torch_cuda
|
torch_cpu -Wl,--no-as-needed
|
c10
|
c10_cuda
|
gomp
|
nvToolsExt
|
opencv_core
|
opencv_highgui
|
#opencv_imgcodecs
|
opencv_imgproc
|
#opencv_videoio
|
opencv_video
|
|
cublas
|
stdc++fs #设置-lstdc++fs 新链接器选项，防止std等gcc库调用不到
|
curand
|
cuda
|
cudnn
|
jsoncpp
|
dl
|
m
|
)
|
|
include_directories(
|
${CMAKE_CURRENT_SOURCE_DIR}
|
${PYTORCH_DIR}/include/
|
${PYTORCH_DIR}/include/torch/csrc/api/include
|
${OPENCV_DIR}/include
|
#/home/disk1/s_opt/opencv/opencv3.4.8/include
|
./
|
)
|
|
cuda_add_executable(${PROJECT_NAME}
|
sdk.cpp
|
main.cpp
|
${SOURCES}
|
)
|
|
target_link_libraries(${PROJECT_NAME}
|
${LIBS}
|
)
|
