TensorRT – 安装TensorRT工具Polygraphy
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文件
在上一步创建的虚拟环境下使用命令:
pip setup.py install
安装polygraphy包。
本文作者:StubbornHuang
版权声明:本文为站长原创文章,如果转载请注明原文链接!
原文标题:TensorRT – 安装TensorRT工具Polygraphy
原文链接:https://www.stubbornhuang.com/1736/
发布于:2021年10月05日 22:22:59
修改于:2023年06月26日 21:13:43
当前分类随机文章推荐
- TensorRT - 使用C++ SDK出现无法解析的外部符号 "class sample::Logger sample::gLogger"错误 阅读688次,点赞0次
- TensorRT - 使用TensorRT C++ SDK部署模型时推理时间波动不稳定或者推理速度越来越慢的问题 阅读285次,点赞0次
- TensorRT - 安装TensorRT工具Polygraphy 阅读5414次,点赞0次
- TensorRT - Using an engine plan file across different models of devices is not recommended and is likely to affect performance or even cause errors 阅读596次,点赞0次
- TensorRT - 使用trtexec工具转换模型、运行模型、测试网络性能 阅读5185次,点赞2次
- TensorRT - 转换onnx模型出现Slice_74 requires bool or uint8 I/O but node can not be handled by Myelin错误 阅读744次,点赞0次
- TensorRT - 使用torch普通算子组合替代torch.einsum爱因斯坦求和约定算子的一般性方法 阅读3508次,点赞3次
- TensortRT - 转换模型出现Could not locate zlibwapi.dll. Please make sure it is in your library path!错误 阅读1204次,点赞0次
- TensorRT - workspace的作用 阅读1353次,点赞0次
- TensorRT - Invalid Node - TopK,This version of TensorRT only supports input K as an initializer 阅读866次,点赞0次
全站随机文章推荐
- 资源分享 - High Dynamic Range Imaging- Acquisition, Display, and Image-Based Lighting ( First Edition )PDF下载 阅读3214次,点赞0次
- 资源分享 - Qt5.9 c++开发指南 PDF下载 阅读24936次,点赞29次
- Windows - 使用类的成员函数作为Win32窗口消息回调处理函数WindowProc 阅读1388次,点赞0次
- 资源分享 - 交互式计算机图形学:基于WebGL的自顶向下方法(第七版),Interactive Computer Graphics - A top-down approach with WebGL(Seven 7th Edition)中文版PDF下载 阅读954次,点赞0次
- OnnxRuntime - 如何部署多个输入和多个输出tensor的onnx模型 阅读200次,点赞0次
- C++ - 获取当前进程内存使用情况 阅读10663次,点赞10次
- Github - Github最简单的下载仓库中单个文件的方法 阅读1223次,点赞0次
- WordPress - get_header函数,加载主题头部header模板 阅读1141次,点赞0次
- OpenCV - 将图片/视频转换为深度学习模型输入格式,BGR通道转RGB,图片归一化,HWC转CHW 阅读5737次,点赞0次
- 深度学习 - Python实现CTC Decode解码算法Greedy Search Decode,Beam Search Decode,Prefix Beam Search Decode 阅读2545次,点赞0次
评论
169