tensorrt windows下简易说明

文章目录


前言

本文针对王鑫宇大神在github上的项目进行简单windows下的简单说明


项目地址:

https://github.com/wang-xinyu/tensorrtx

一、使用流程

1.创建一个控制台程序。

最好是选择release版本+x64这个组合。release是个人习惯,以前被libtorch伤过,debug版本实在太慢太慢太慢了,release的代码也可以调试,改一下C/C++中的优化(已禁用),生成调试信息中选(生成调试信息(DEBUG))

tensorrt windows下简易说明
tensorrt windows下简易说明

2.将所有的.cpp文件和.h文件加入进工程。

.cu文件也放到源文件中。

tensorrt windows下简易说明

3.接下来就是一些库文件(.h)和链接文件(.lib)目录的编辑工作了。我这里给出大家我的路径:
tensorrt windows下简易说明

1.CUDA的include路径。
2.opencv的include路径。
3.tensorrt的include路径。
4.通常有时候有一些附加头文件的时候我们可以放在自己工程里,也可以自己新建一个inc文件夹,将这个文件夹的路径包含进来。

tensorrt windows下简易说明

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啥的错误。从这里找一下生成自定义选项,如果没出现试着先编译一下工程就出来了。
tensorrt windows下简易说明
tensorrt windows下简易说明
选自己用的CUDA版本就好。
另一个就是程序中报这个:
tensorrt windows下简易说明
在预处理器中加两个宏定义试试看:
tensorrt windows下简易说明
tensorrt windows下简易说明
再编译试试看,一般都可以成功。

上一篇:从App Store 获取ipa 文件


下一篇:Android NDK开发入门基础