cmake_minimum_required(VERSION 3.24)
project(test01)
set(CMAKE_CXX_STANDARD 17)
#添加include路径
include_directories(/home/qf/TensorRT_8.2.5.1/include)
include_directories(/home/qf/TensorRT_8.2.5.1/samples/common)
#添加.so库文件路径 相当于 -L
link_directories(/usr/local/cuda-11.4/targets/x86_64-linux/lib) #添加动态连接库的路径
link_directories(/home/qf/TensorRT_8.2.5.1/targets/x86_64-linux-gnu/lib)
#添加一个lib文件下所有so文件
set(TENSORRT_LIB_PATH “/home/qf/TensorRT_8.2.5.1/lib”)
file(GLOB LIBS “${TENSORRT_LIB_PATH}/*.so”)
#快捷添加CUDA
find_package(CUDA 11.4 REQUIRED)
message(“CUDA_LIBRARIES:${CUDA_LIBRARIES}”)
message(“CUDA_INCLUDE_DIRS:${CUDA_INCLUDE_DIRS}”)
include_directories(${CUDA_INCLUDE_DIRS})
#总链接操作 -lcudart
add_executable(test01 main.cpp)
target_link_libraries(test01 ${LIBS} ${CUDA_LIBRARIES} cudart nvinfer )
clion 配置 cmakelist.txt
最新推荐文章于 2023-01-31 20:34:23 发布