栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > C/C++/C#

【pytorch】Win10安装C++版及python版本tensorRT

C/C++/C# 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

【pytorch】Win10安装C++版及python版本tensorRT

tensorRT提供了C++及Python的API接口。

(一)安装Python版接口
linux安装较为简介,直接使用:

pip install tensorrt
pip install nvidia-pyindex
pip install nvidia-tensorrt

进行安装即可。
对于windows版本tensorRT:
登录英伟达官网下载对应版本的压缩包:
下载链接
下载后打开目录,在文件夹位置下:
D:TensorRT-8.4.0.6.Windows10.x86_64.cuda-11.6.cudnn8.3TensorRT-8.4.0.6python
这里使用的是python3.9,所以使用的安装包是:
tensorrt-8.4.0.6-cp39-none-win_amd64.whl

1.在系统环境变量中新增:
‪D:TensorRT-8.4.0.6.Windows10.x86_64.cuda-11.6.cudnn8.3TensorRT-8.4.0.6lib
2.安装graphsurgeon
D:TensorRT-8.4.0.6.Windows10.x86_64.cuda-11.6.cudnn8.3TensorRT-8.4.0.6graphsurgeon> pi
p install graphsurgeon-0.4.5-py2.py3-none-any.whl
3.安装onnx_graphsurgeon
 D:TensorRT-8.4.0.6.Windows10.x86_64.cuda-11.6.cudnn8.3TensorRT-8.4.0.6onnx_graphsurgeo
n> pip install onnx_graphsurgeon-0.3.12-py2.py3-none-any.whl
4.安装 uff
D:TensorRT-8.4.0.6.Windows10.x86_64.cuda-11.6.cudnn8.3TensorRT-8.4.0.6uff> pip install
 uff-0.6.9-py2.py3-none-any.whl
5.安装tensorRT:
  D:TensorRT-8.4.0.6.Windows10.x86_64.cuda-11.6.cudnn8.3TensorRT-8.4.0.6python> pip inst
all tensorrt-8.4.0.6-cp39-none-win_amd64.whl 

将lib文件夹下的dll文件复制进cudnn文件夹下。

import tensorrt as trt
print(trt.__version__)

输出错误信息:

Note: Paths searched were:
['D:\anaconda\envs\mypytorch\Lib\site-packages\cv2\../../x64/vc14/bin', 'D:\anaconda\envs\mypytorch'

选择其中一个路径,将libs文件夹下的若干dll,复制到其中一个目录。
至此安装完成。程序输出:

8.4.0.6

注:一般修改系统变量后,需重启IDE,如pycharm,可使系统变量生效。如果缺少dll,可用everything搜索工具进行检索,再将对应dll路径添加进系统变量内。
附:安装tensorRT所需的pycuda库:
找到合适的pycuda的版本:
下载后进行安装即可:
pip install D:pycuda-2021.1+cuda115-cp39-cp39-win_amd64.whl
若出现类似:
RuntimeError: module compiled against numpy version 0xf but this version of numpy is 0xd
直接 pip uninstall numpy并重装。

(一)安装C++版接口
为正常使用C++版接口,需去英伟达官网下载安装NVIDIA GPU Computing Toolkit
使用其自带的VS样例做验证:D:TensorRT-8.4.0.6.Windows10.x86_64.cuda-11.6.cudnn8.3TensorRT-8.4.0.6samplessampleMNIST

打开该VS项目,在项目的库路径中添加CUDA路径:

并将D:TensorRT-8.4.0.6.Windows10.x86_64.cuda-11.6.cudnn8.3TensorRT-8.4.0.6lib下的所有dll放入exe生成目录。安装完成,编译运行得到如下结果:

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/873298.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号