-
Notifications
You must be signed in to change notification settings - Fork 0
/
CMakeLists-win.txt
99 lines (84 loc) · 3.3 KB
/
CMakeLists-win.txt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
cmake_minimum_required(VERSION 3.28)
project(yolov5_trtx_v10)
set(CMAKE_CXX_STANDARD 17)
# 设置nvcc编译cu文件时候使用utf-8编码
set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -Xcompiler /utf-8")
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} /utf-8")
enable_language(CUDA)
# 设置cuda多个框架支持
set(CMAKE_CUDA_ARCHITECTURES 75 86 89)
message(STATUS "CMAKE_CUDA_ARCHITECTURES: ${CMAKE_CUDA_ARCHITECTURES}")
# OpenCV
set(OpenCV_DIR E:\\Opencv\\install\\opencv-4.8.0\\build)
find_package(OpenCV REQUIRED)
include_directories(${OpenCV_INCLUDE_DIRS})
link_directories(${OpenCV_LIB_DIR})
# CUDA
set(CUDA_TOOLKIT_ROOT_DIR C:\\Program\ Files\\NVIDIA\ GPU\ Computing\ Toolkit\\CUDA\\v11.8)
include_directories(${CUDA_TOOLKIT_ROOT_DIR}/include)
link_directories(${CUDA_TOOLKIT_ROOT_DIR}/lib/x64)
# TensorRT
#set(TENSORRT_ROOT E:\\TensorRT\\TensorRT-8.6.1.6)
set(TENSORRT_ROOT E:\\TensorRT\\TensorRT-10.2.0.19)
include_directories(${TENSORRT_ROOT}/include)
link_directories(${TENSORRT_ROOT}/lib)
# 判断TENSORRT_ROOT路径中的version如果路径中第一个.前大于8
# 获取所有版本文件
file(GLOB TENSORRT_VERSION_FILES "${TENSORRT_ROOT}/include/NvInferVersion.h")
# 读取版本文件
file(STRINGS ${TENSORRT_VERSION_FILES} TENSORRT_VERSION_LINES
LIMIT_COUNT 1 # 只读取第一行
REGEX "#define NV_TENSORRT_MAJOR [0-9]+" # 匹配版本号定义行
)
message(STATUS " TENSORRT_VERSION_LINES: ${TENSORRT_VERSION_LINES}")
# 解析版本号
string(REGEX REPLACE "#define NV_TENSORRT_MAJOR ([0-9]+)" "\\1" TENSORRT_VERSION_MAJOR ${TENSORRT_VERSION_LINES})
message(STATUS " TENSORRT_VERSION_MAJOR: ${TENSORRT_VERSION_MAJOR}")
# 判断版本号是否大于等于10
if (TENSORRT_VERSION_MAJOR GREATER_EQUAL 10)
message(STATUS " TensorRT version is greater than or equal to 10.")
link_libraries(
opencv_core
opencv_highgui
opencv_imgproc
opencv_imgcodecs
cudart
cublas
nvinfer_10
nvinfer_plugin_10
)
else ()
message(STATUS " TensorRT version is less than 10.")
link_libraries(
opencv_core
opencv_highgui
opencv_imgproc
opencv_imgcodecs
cudart
cublas
nvinfer
nvinfer_plugin
)
endif ()
include_directories(${CMAKE_SOURCE_DIR}/include)
include_directories(${CMAKE_SOURCE_DIR}/plugin)
include_directories(${CMAKE_SOURCE_DIR}/src)
link_directories(${CMAKE_SOURCE_DIR}/lib)
add_definitions(-DNOMINMAX)
add_definitions(-DAPI_EXPORTS)
file(GLOB_RECURSE SRCS ${CMAKE_SOURCE_DIR}/src/*.cpp ${CMAKE_SOURCE_DIR}/src/*.cu)
file(GLOB_RECURSE PLUGIN_SRCS ${PROJECT_SOURCE_DIR}/plugin/*.cu)
add_library(myplugins SHARED ${PLUGIN_SRCS})
target_link_libraries(myplugins nvinfer_10 cudart)
add_executable(yolov5_cls yolov5_cls.cpp ${SRCS})
target_link_libraries(yolov5_cls myplugins)
add_executable(yolov5_det yolov5_det.cpp ${SRCS})
target_link_libraries(yolov5_det nvinfer_10)
target_link_libraries(yolov5_det cudart)
target_link_libraries(yolov5_det myplugins)
target_link_libraries(yolov5_det ${OpenCV_LIBS})
add_executable(yolov5_seg yolov5_seg.cpp ${SRCS})
target_link_libraries(yolov5_seg nvinfer_10)
target_link_libraries(yolov5_seg cudart)
target_link_libraries(yolov5_seg myplugins)
target_link_libraries(yolov5_seg ${OpenCV_LIBS})