TensorRT – 安装TensorRT工具Polygraphy
本文作者:StubbornHuang
版权声明:本文为站长原创文章,如果转载请注明原文链接!
原文标题:TensorRT – 安装TensorRT工具Polygraphy
原文链接:https://www.stubbornhuang.com/1736/
发布于:2021年10月05日 22:22:59
修改于:2021年10月05日 22:22:59

1 项目地址
Github地址:https://github.com/NVIDIA/TensorRT/tree/master/tools/Polygraphy
2 官方步骤
2.1 官方的安装步骤(未成功)
按官方的步骤没有成功,可能是我Anaconda的网络环境没有配置好。
2.1.1 从线上安装预处理包
python -m pip install colored polygraphy --extra-index-url https://pypi.ngc.nvidia.com
2.1.2 手动安装
详情请参考官方文档
2.2 自己的安装步骤(成功)
2.2.1 创建conda虚拟环境
先创建一个conda虚拟环境
conda create -n Polygrapgy python=3.7
2.2.2 手动使用Polygraphy的setup.py安装
先迁移整个TensorRT仓库:https://github.com/NVIDIA/TensorRT
然后进入仓库polygraphy工具所在目录:TensorRT-master\tools\Polygraphy,在该目录下有setup.py文件

在上一步创建的虚拟环境下使用命令:
pip setup.py install
安装polygraphy包。
3 polygraphy相关依赖库安装
3.1 自动安装依赖
官方文档上指明通过设置POLYGRAPHY_AUTOINSTALL_DEPS=1或者在代码中加上polygraphy.config.AUTOINSTALL_DEPS = True即可自动下载依赖库。
pycharm中设置环境变量可参考下图:

3.2 手动安装依赖
官方文档指出在仓库TensorRT-master\tools\Polygraphy\polygraphy\backend下每一个文件夹都有requirements.txt,可以通过以下的方式一个一个文件夹手动安装依赖项:
python -m pip install -r polygraphy/backend/<name>子文件夹名/requirements.txt
我将每个文件夹的requirements.txt都看了一下,主要的依赖库如下:
- onnx
- onnxruntime
- numpy
- onnx_graphsurgeon
- torch
- tensorflow
在安装的时候需要先安装onnxruntime 再安装onnxruntime-gpu=1.1.2,不然会出现找不到onnx相关module的问题。
当前分类随机文章推荐
- TensorRT - 喜大普奔,TensorRT8.2 EA起开始支持Einsum爱因斯坦求和算子 阅读916次,点赞0次
- TensorRT - Polygraphy工具的使用 阅读1960次,点赞0次
- TensorRT - 使用trtexec工具转换模型、运行模型、测试网络性能 阅读1775次,点赞0次
- TensorRT - Windows下TensorRT下载与配置 阅读982次,点赞0次
- TensorRT - 扩展TensorRT C++API的模型输入维度,增加Dims5,Dims6,Dims7,Dims8 阅读936次,点赞0次
- TensorRT - 解决INVALID_ARGUMENT: getPluginCreator could not find plugin ScatterND version 1,TensorRT找不到ScatterND插件的问题 阅读1480次,点赞0次
- TensorRT - 使用torch普通算子组合替代torch.einsum爱因斯坦求和约定算子的一般性方法 阅读1196次,点赞0次
- TensorRT - 自带工具trtexec的参数使用说明 阅读2058次,点赞0次
- TensorRT - 安装TensorRT工具Polygraphy 阅读1676次,点赞0次
全站随机文章推荐
- 资源分享 - OpenGL Insights 英文高清PDF下载 阅读1692次,点赞0次
- 三维旋转 - 欧拉角和旋转矩阵的基本概念以及相互转换 阅读310次,点赞0次
- 资源分享 - Game AI Pro 360 - Guide to Movement and Pathfinding 英文高清PDF下载 阅读1111次,点赞0次
- Duilib - 超链接文本 阅读2375次,点赞0次
- 资源分享 - C++程序设计-原理与实践(基础篇),第2版 高清PDF下载 阅读1122次,点赞0次
- Python - 解决opencv-python使用cv2.imread()读取中文路径图片失败的问题 阅读43次,点赞0次
- C++ - 字节数组byte[]或者unsigned char[]与double的相互转换 阅读439次,点赞0次
- C++ – UTF8编码下的全角字符转半角字符 阅读622次,点赞0次
- Windows平台录音类封装:AudioRecordWindows 阅读3059次,点赞0次
- Modern OpenGL - 与着色器Shader相关的API函数总结以及如何创建、编译与使用Shader 阅读63次,点赞0次
评论
148