Win10+TensorRT 8.5安装+VS2022配置

环境
  • Anaconda3 [python 3.8]
  • Visual Studio 2022
  • CUDA 11.8
  • cuDNN 8.6.0
  • TensorRT 8.5.2.2

TensorRT 下载并 配置
# 如下载8.5.2.2版本
TensorRT-8.5.2.2.Windows10.x86_64.cuda-11.8.cudnn8.6.zip
  • 加压后,添加环境变量PATH中(根据自己路径)
# 将lib路径添加到环境变量
D:\AI_SoftEnv\TensorRT\TensorRT-8.5.2.2\lib
  • 将TensorRT-8.5.2.2\lib下的lib文件和dll文件分别拷贝到CUDA的lib/x64路径和bin路径下
 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\bin
 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\lib\x64

在这里插入图片描述

  • 安装pytho版本tensorrt
 # 到TensorRT-8.5.2.2\python下,根据自己的python版本安装tensorrt
 pip install tensorrt-8.5.2.2-cp38-none-win_amd64.whl
 
 # 测试
 import tensorrt as trt
 trt.__version__

VS2022配置TensorRT并测试

使用VS2022打开TensorRT-8.5.2.2\samples\sampleOnnxMNIST\sample_onnx_mnist.sln
在这里插入图片描述

进行如下环境配置
在这里插入图片描述

在这里插入图片描述
在这里插入图片描述
注:对于VS2022,其已经默认添加了这几个库(前4个是tensorrt的,后3个是cuda的)。为了说明,还是按照上面方式配置即可。

nvinfer.lib
nvinfer_plugin.lib
nvonnxparser.lib
nvparsers.lib
cudnn.lib
cublas.lib
cudart.lib

到此为止,TensorRT在VS中就配置好了(实际上面也展示了CUDA的配置),下面就可以直接测试案例程序了。

为了防止后面执行exe操作闪退,添加getchar();到main函数下
在这里插入图片描述

之后编译build即可。
编译成功后,会在 TensorRT-8.5.2.2\bin 路径下生成可执行文件sample_onnx_mnist.exe
在这里插入图片描述
双击运行即可
在这里插入图片描述


以上就是TensorRT安装和VS配置以及官方Demo运行过程,到此就说明一切都OK了。


备注

以上过程可能存在找不到zlibwapi.dll的问题,解决办法:

  • 官网下载 ZLIB DLL 库,解压后将zlibwapi.dll 拷贝到C:\Windows\System32下即可。
    也可分别将zlibwapi.dllzlibwapi.lib放于CUDA的bin和lib文件夹下,两种方法都行。
# zlibwapi.dll放于其中
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\bin 
# zlibwapi.lib放于其中
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\lib

还有一个关于懒加载问题,来源和配置办法如下
  • 来源

在这里插入图片描述
根据官方说明,需要配置环境变量
在这里插入图片描述


参考

https://blog.csdn.net/qq_44224801/article/details/125525721
https://blog.csdn.net/Chaos_Happy/article/details/124064428
https://www.cnblogs.com/xixixing/p/16140973.html
https://www.iotword.com/5928.html

  • 19
    点赞
  • 87
    收藏
    觉得还不错? 一键收藏
  • 22
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 22
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值