360网站收录提交入口,房地产销售技巧,软件工程最好的出路,企业网站排名提升软件为避免重复#xff0c;一些安装内容我直接贴其他大佬的帖子了#xff0c;我是按照他们的步骤来操作的#xff0c;趟过一遍#xff0c;没有问题。 本篇着重在tensort在Cmakelist中如何配置#xff0c;以及如何配置编译动/静态库#xff0c;比较基础#xff0c;也是想做个… 为避免重复一些安装内容我直接贴其他大佬的帖子了我是按照他们的步骤来操作的趟过一遍没有问题。 本篇着重在tensort在Cmakelist中如何配置以及如何配置编译动/静态库比较基础也是想做个笔记记录一下。 文章目录 一、环境二、安装cuda和cudnn三、安装tensorrt以及配置四、CMakeLists如何配置tensorrt以及坑 一、环境
windows windows11cuda11.8cudnn8注意和cuda配套tensorrt8.6.1.6 linux ubuntu18.04 x86cuda: 11.4cudnn: 8tensorrt8.6.1.6
二、安装cuda和cudnn
linux可以参考ubunu安装cuda
三、安装tensorrt以及配置
均可以参考https://blog.csdn.net/djstavaV/article/details/125195569
四、CMakeLists如何配置tensorrt以及坑
这里的场景是这样的有两个工程A工程来调用B工程B工程中包含tensorrt的推理操作依赖tensorrt那么这就需要把B工程打包成库文件关于Cmakelist详细指导可以跳转到如何在CMakelist中配置这里编译成库文件在Cmakelist中需要把
add_executable(${PROJECT_NAME} ${SOURCE_FILES})改成
add_library(${PROJECT_NAME} STATIC ${SOURCE_FILES}) 注意 这里一定要用静态库不然会报这个错误
ERROR: 1: [dispatchStubs.cpp::deserializeEngine::14] Error Code 1: Internal Error (Unexpected call to stub)查询资料参考https://github.com/NVIDIA/TensorRT/issues/3307
然后要注意顺序
include_directories - link_directories - add_library - target_link_libraries - install对了这里最后一步最好install一下可以加入到系统路径/usr/local/lib/中。可参考
install(TARGETS ${PROJECT_NAME} LIBRARY DESTINATION /usr/local/lib/)其他的target_link_libraries把tensorrt依赖的库文件放入即可包括cuda的这里也举例
target_link_libraries(${PROJECT_NAME} cublas cudart cudadevrt nvinfer nvinfer_dispatch nvinfer_lean nvinfer_plugin nvinfer_vc_plugin nvonnxparser nvparsers)然后include_directories和link_directories分别放入cuda和tensorrt的include文件和lib文件即可。 注意 include、lib和链接库一定要加不然会出现以下问题
cmakelist /usr/bin/ld: cannot find -lcublas小抄 add_executable 用来生成可执行文件add_library 用来生成库文件其中的关键字SHARED和STATIC又分别代表动态库和静态库。
以上就是全部内容有什么问题可以评论一起交流 Enjoy~
