1 项目地址

Github地址:https://github.com/NVIDIA/TensorRT/tree/master/tools/Polygraphy

2 官方步骤

2.1 官方的安装步骤

2.1.1 预处理包

polygraphy依赖以下包:

  • onnx
  • onnxruntime
  • onnx_graphsurgeon

依次使用以下命令安装依赖库onnx和onnxruntime:

pip install onnx
pip install onnxruntime

然后安装onnx_graphsurgeon

pip install onnx_graphsurgeon --index-url https://pypi.ngc.nvidia.com

最后安装polygraphy

pip install colored polygraphy --extra-index-url https://pypi.ngc.nvidia.com

安装完成之后就可以在命令行使用polygraphy工具了。

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文件

TensorRT – 安装TensorRT工具Polygraphy-StubbornHuang Blog

在上一步创建的虚拟环境下使用命令:

pip setup.py install

安装polygraphy包。