文章目录
前言
本文针对王鑫宇大神在github上的项目进行简单windows下的简单说明
项目地址:
https://github.com/wang-xinyu/tensorrtx
一、使用流程
1.创建一个控制台程序。
最好是选择release版本+x64这个组合。release是个人习惯,以前被libtorch伤过,debug版本实在太慢太慢太慢了,release的代码也可以调试,改一下C/C++中的优化(已禁用),生成调试信息中选(生成调试信息(DEBUG))
2.将所有的.cpp文件和.h文件加入进工程。
.cu文件也放到源文件中。
3.接下来就是一些库文件(.h)和链接文件(.lib)目录的编辑工作了。我这里给出大家我的路径:
1.CUDA的include路径。
2.opencv的include路径。
3.tensorrt的include路径。
4.通常有时候有一些附加头文件的时候我们可以放在自己工程里,也可以自己新建一个inc文件夹,将这个文件夹的路径包含进来。
1.tensorrt的library路径。
2.CUDA的library路径。
3.opencv的library路径。
而附加库一般包含的是这些:
nvinfer.lib
nvinfer_plugin.lib
cudart.lib
cudadevrt.lib
cudart_static.lib
kernel32.lib
user32.lib
gdi32.lib
winspool.lib
shell32.lib
ole32.lib
oleaut32.lib
uuid.lib
comdlg32.lib
advapi32.lib
opencv_world440.lib
自己记着修改一下自己opencv library名称,不同人的opencv版本是不一样的,建议使用world版本的opencv,使用起来方便一些。
至此,你可以试着编译一下,发现是不是没成果,别着急,一般情况下有几个可能性,一个就是报CUDA啥的错误。从这里找一下生成自定义选项,如果没出现试着先编译一下工程就出来了。
选自己用的CUDA版本就好。
另一个就是程序中报这个:
在预处理器中加两个宏定义试试看:
再编译试试看,一般都可以成功。