1. 根据CUDA版本到网站下载TensorRT5,解压会得到一个TensorRT-5.0.2.6
2. 新建Clion工程,会看到CMakelist.txt文件
3. 进行这些配置
#TensorRT5
include_directories(/home/dwy/TensorRT5/include)
include_directories(/home/dwy/TensorRT5/samples/common)
set(TENSORRT_LIB_PATH “/home/dwy/TensorRT5/lib”)
file(GLOB LIBS “${TENSORRT_LIB_PATH}/*.so”)
#CUDA
find_package(CUDA 9.0 REQUIRED)
message(“CUDA_LIBRARIES:${CUDA_LIBRARIES}”)
message(“CUDA_INCLUDE_DIRS:${CUDA_INCLUDE_DIRS}”)
include_directories(${CUDA_INCLUDE_DIRS})
#和文件连在一起
add_executable(sampleMNISTAPI sampleMNISTAPI.cpp)
target_link_libraries(sampleMNISTAPI ${LIBS} ${CUDA_LIBRARIES})
TensorRT5在Clion C++中的安装配置
最新推荐文章于 2024-05-28 14:47:52 发布