tensorrt windows下简易说明

2021/6/13 7:25:25

本文主要是介绍tensorrt windows下简易说明,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

文章目录

  • 前言
  • 一、使用流程


前言

本文针对王鑫宇大神在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版本就好。
另一个就是程序中报这个:
在这里插入图片描述
在预处理器中加两个宏定义试试看:
在这里插入图片描述
在这里插入图片描述
再编译试试看,一般都可以成功。



这篇关于tensorrt windows下简易说明的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!


扫一扫关注最新编程教程