TensorRT安装

TensorRT是可以在NVIDIA各种GPU硬件平台下运行的推理引擎,是一个高性能的深度学习推理优化器,可以为深度学习的落地应用提供低延迟、高吞吐率的部署推理。由C++、CUDA、python三种语言编写成的一个库,其中核心代码的语言为C++和CUDA,但是它也有python的api接口。

🎏目录

    🎈1 TensorRT安装
      🎄1.1 下载
      🎄1.2 配置
      🎄1.3 库安装
    🎈2 测试
      🎄2.1 VS2022配置
      🎄2.2 缺少windows sdk问题
      🎄2.3 生成工具问题
      🎄2.4 zlibwapi.dll问题
      🎄2.5 验证

✨1 安装TensorRT

🎈1.1 下载

1️⃣TensorRT下载地址
一定要选择对应自己CUDA版本的,比如我的CUDA11.8
在这里插入图片描述
我选择了8.6,下载完成后,压缩到指定路径。

🍕1.2 配置

配置与cudnn安装类似:

  • TensorRT-7.2.2.3\include头文件copy到C:\Program Files\NVIDIA GPU ComputingToolkit\CUDA\v10.2\include
  • TensorRT-7.2.2.3\lib中所有lib文件copy到C:\Program Files\NVIDIA GPU
    ComputingToolkit\CUDA\v10.2\lib\x64
  • TensorRT-7.2.2.3\lib中所有dll文件copy 到C:\Program Files\NVIDIA GPU
    ComputingToolkit\CUDA\v10.2\bin

同时将lib路径添加到环境变量

🍔1.3 tenosrrt库安装

进入tensorRT解压路径下的python目录内,利用pip安装与python版本对应的whl文件的tensorrt。
比如,python版本3.9,目录下whl文件
在这里插入图片描述
那么,就应该是pip install tensorrt-8.5.3.1-cp39-none-win_amd64.whl

测试是否安装成功用:

 import tensorrt as trt
 trt.__version__

✨2 测试

🌭2.1 VS2022配置

  1. 利用vs打开tensorrt目录下的samples\sampleOnnxMNIST\sample_onnx_mnist.sln,进行环境配置。
    点击一下右侧箭头指向,并找到项目=》属性并打开
    在这里插入图片描述
  2. 找到C/C++=》附加包含目录,打开添加tensorrt的include路径
    在这里插入图片描述
  3. 找到链接器=》常规=》附加库目录,添加tensorrt的lib路径
    在这里插入图片描述
  4. 找到链接器=》输入=》附加依赖项,添加路径TensorRT解压路径\lib\*.lib
    在这里插入图片描述
  5. 为了防止后面执行exe操作闪退,添加getchar();到main函数下
    在这里插入图片描述
  6. 开始编译运行
    在这里插入图片描述

✨2.2 缺少windows sdk问题

最后一步编译时,遇到的第一个错误没有对应的版本的windos sdk
在官网https://developer.microsoft.com/zh-cn/windows/downloads/sdk-archive/找到合适的sdk版本并安装(点击安装sdk会有个exe文件线上安装)

🎃2.3 生成工具问题

安装完成后出现了没有2017年生成工具

  1. 找到电脑上的Visual Studio Installer,打开
  2. 点击修改
    在这里插入图片描述
  3. 找到2017年的生成工具勾选,点击修改进行安装
    在这里插入图片描述
    其余版本也一致。

🎄2.4zlibwapi.dll问题

提示问题,找不到zlibwapi.dll

  1. 进入官网下载Zlib
    在这里插入图片描述
  2. 将下载的压缩包中的zlibwapi.dllzlibwapi.lib放于CUDA的bin和lib文件夹下
    在这里插入图片描述
    在这里插入图片描述

🎆2.5 验证

按照上面的办法处理完错误后,再次编译,便成功完成了。
后面会在TensorRT的bin文件夹下生成sample_onnx_mnist.exe,双击打开
在这里插入图片描述
呈现类似的结果就是成功了

  • 3
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

白三点

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值