62 lines
1.5 KiB
CMake
62 lines
1.5 KiB
CMake
cmake_minimum_required(VERSION 3.10)
|
|
cmake_policy(SET CMP0079 NEW)
|
|
|
|
project(cuda_infer LANGUAGES CXX CUDA)
|
|
|
|
# 设置标准
|
|
set(CMAKE_CXX_STANDARD 17)
|
|
set(CMAKE_CXX_STANDARD_REQUIRED ON)
|
|
set(CMAKE_CUDA_STANDARD 17)
|
|
set(CMAKE_CUDA_STANDARD_REQUIRED ON)
|
|
set(CUDA_USE_STATIC_CUDA_RUNTIME OFF)
|
|
set(CMAKE_BUILD_TYPE "Release")
|
|
# 抑制过时 API 警告
|
|
add_compile_options(-Wno-deprecated-declarations)
|
|
|
|
# 禁用 .rsp 响应文件（避免 nvcc 报错）
|
|
set(CMAKE_CUDA_USE_RESPONSE_FILE_FOR_OBJECTS OFF)
|
|
set(CMAKE_CUDA_USE_RESPONSE_FILE_FOR_INCLUDES OFF)
|
|
|
|
# 查找依赖
|
|
find_package(CUDAToolkit REQUIRED)
|
|
find_package(OpenCV REQUIRED)
|
|
find_package(Eigen3 REQUIRED)
|
|
|
|
# 收集源码
|
|
file(GLOB_RECURSE CUDA_INFER_SRC
|
|
${CMAKE_CURRENT_SOURCE_DIR}/*.cpp
|
|
${CMAKE_CURRENT_SOURCE_DIR}/*.cu
|
|
)
|
|
|
|
# 添加静态库
|
|
add_library(cuda_infer STATIC ${CUDA_INFER_SRC})
|
|
set_target_properties(cuda_infer PROPERTIES POSITION_INDEPENDENT_CODE ON)
|
|
|
|
# 设置包含路径
|
|
target_include_directories(cuda_infer PUBLIC
|
|
${CMAKE_CURRENT_SOURCE_DIR}
|
|
${CUDAToolkit_INCLUDE_DIRS}
|
|
${TensorRT_INCLUDE_DIR}
|
|
${EIGEN3_INCLUDE_DIRS}
|
|
${OpenCV_INCLUDE_DIRS}
|
|
)
|
|
|
|
# 设置 CUDA 编译选项
|
|
target_compile_options(cuda_infer PRIVATE
|
|
$<$<COMPILE_LANGUAGE:CUDA>:
|
|
--generate-code=arch=compute_86,code=sm_86
|
|
-Xcompiler=-fPIC
|
|
-O3
|
|
-w
|
|
-Wno-deprecated-gpu-targets
|
|
-Wno-error=deprecated-declarations
|
|
>
|
|
)
|
|
|
|
# 链接库
|
|
target_link_libraries(cuda_infer PRIVATE
|
|
${OpenCV_LIBS}
|
|
CUDA::cudart
|
|
TensorRT::TensorRT
|
|
)
|