详解Windows系统安装TensorRT

文章详细介绍了在Windows系统中安装TensorRT的步骤,包括下载TensorRT、CUDA和CUDNN,以及如何解压安装文件、复制到相应目录和安装Python相关whl包。在安装过程中需要注意与CUDA版本的匹配,并且在遇到导入问题时,可能需要更新系统环境变量。最后提到了测试阶段可能遇到的问题及其解决方案。

TensorRT 是 NVIDIA 推出的一款高性能神经网络部署引擎.Windows系统下TensorRT目前不能简单直接通过pip指令自动下载安装, 安装之前还需要提前安装好 CUDACUDNN.

CUDA和CUDNN安装可参考:

下载TensorRT

  1. 进入NVIDIA的TensorRT页面: https://developer.nvidia.com/tensorrt. 点击Download now, 进入TensorRT下载页面. PS: TensorRT页面需要先登录NVIDIA账号, 才会有下载选项.

在这里插入图片描述

  1. 选择版本, 这里博主选择最新的TensorRT 8.
    在这里插入图片描述
  2. 勾选"I Agree To the Terms of the NVIDIA TensorRT License Agreement"
    在这里插入图片描述
  3. 选择具体版本, 这里博主选择的是TensorRT 8.5 GA Update 2.
    在这里插入图片描述
  4. 选择Zip Packages for Windows下面和自己CUDA版本匹配的ZIP, 点击即可下载.
    在这里插入图片描述

安装TensorRT

  1. 解压下载的ZIP文件, 如下:
    在这里插入图片描述
  2. 从解压好的文件夹中复制文件到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6, 对应复制路径. PS: 此处的复制需要管理员权限. 此处CUDA的版本v11.6需要根据自身安装的情况而定.
序号所需复制文件来源目标
1…\TensorRT-8.5.3.1\bin\trtexec.exe…\TensorRT-8.5.3.1\binC:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\bin
2…\TensorRT-8.5.3.1\include下的所有文件…\TensorRT-8.5.3.1\includeC:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\include
3🌟…\TensorRT-8.5.3.1\lib下的所有lib文件…\TensorRT-8.5.3.1\libC:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\lib\x64
4🌟…\TensorRT-8.5.3.1\lib下的所有dll文件…\TensorRT-8.5.3.1\libC:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\lib
  1. 安装whl文件. 进入文件所在目录, 打开终端, 调用指令安装. PS: 安装时一定要注意whl文件的安装路径. 建议直接使用cd命令, 使终端进入到对应文件目录, 再使用 pip install *.whl 安装.
序号待安装whl文件所在文件夹安装指令
1graphsurgeon-0.4.6-py2.py3-none-any.whl…\TensorRT-8.5.3.1\graphsurgeonpip install graphsurgeon-0.4.6-py2.py3-none-any.whl
2onnx_graphsurgeon-0.3.12-py2.py3-none-any.whl…\TensorRT-8.5.3.1\onnx_graphsurgeonpip install onnx_graphsurgeon-0.3.12-py2.py3-none-any.whl
3uff-0.6.9-py2.py3-none-any.whl…\TensorRT-8.5.3.1\uffpip install uff-0.6.9-py2.py3-none-any.whl
4🌟tensorrt-8.5.3.1-cp38-none-win_amd64.whl 需根据自身python版本选择对应的文件…\TensorRT-8.5.3.1\pythonpip install tensorrt-8.5.3.1-cp38-none-win_amd64.whl

测试

在这里插入图片描述

可能出现的问题

  • import tensorrt时显示找不到nvinfer.dll或者nvparsers.dll等
    解决办法:nvinfer.dll或者nvparsers.dll等dll文件时我们解压tensorrt压缩包时复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\lib文件夹下面的文件,因此只需要将这个文件夹地址添加到系统的环境变量Path中即可。

参考资料

  1. NVIDIA Deep Learning TensorRT Documentation🚀🚀🚀

收集整理和创作不易, 若有帮助🉑, 请帮忙点赞👍➕收藏❤️, 谢谢!✨✨🚀🚀

评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值