TensorRT8.4.x Python API安装配置与测试
OpenCV学堂
发布于 2022-04-18 03:48:04
发布于 2022-04-18 03:48:04
代码可运行
运行总次数：0
代码可运行
关联问题
换一批
点击上方蓝字关注我们
微信公众号：OpenCV学堂 关注获取更多计算机视觉与深度学习知识
版本信息
首先必须要交代一下硬件跟依赖包的版本信息
-GPU -3050ti-Win10 x64-CUDA11.0.1-cuDNN8.2.x-TensorRT8.4.0.x-Python3.6.5-Pytorch1.7.1然后我还下载了YOLOv5的最新版本，并测试通过如下：
Python API配置支持
我把tensorRT解压在D:\TensorRT-8.4.0.6
目录结果如下：
首先输入下面的命令行：
cd /d D:\TensorRT-8.4.0.6到tensorRT文件夹下面，然后分别执行
安装tensorrt python sdk
cd pythonpython.exe -m pip install tensorrt-8.4.0.6-cp36-none-win_amd64.whl注意：一定要跟安装跟你Python SDK对应版本的whl包。
安装onnx python sdk支持
cd..cd onnx_graphsurgeonpython.exe -m pip install onnx_graphsurgeon-0.3.12-py2.py3-none-any.whl安装ONNX格式模型文件读取支持
如果是tensorflow，下面两个也执行一波！
cd..cd graphsurgeonpython.exe -m pip install graphsurgeon-0.4.5-py2.py3-none-any.whlcd..cd uffpython.exe -m pip install uff-0.6.9-py2.py3-none-any.whl这样就安装好啦，别着急，把tensorRT的lib目录配置到系统的环境变量中去：
D:\TensorRT-8.4.0.6\lib然后重新启动cmd，输入如下命令行：
表明TensorRT Python API 已经完全安装成功啦！
测试YOLOv5加速
YOLOv5最新版本中首先使用下面的命令行导出一个tensorRT的engine文件，命令行如下：
python export.py --weights yolov5s.pt --include engine --device 0
然后使用导出的engine测试结果如下：
python detect.py --source data/images/Boogie_Up.mp4 --weights yolov5s.engine运行结果如下：
从此我又相信TensorRT+YOLOv5了
扫码查看OpenCV+Pytorch系统化学习路线图
推荐阅读
OpenCV4.5.4 人脸检测+五点landmark新功能测试
OpenCV4.5.x DNN + YOLOv5 C++推理
OpenCV4.5.4 直接支持YOLOv5 6.1版本模型推理
OpenVINO2021.4+YOLOX目标检测模型部署测试
比YOLOv5还厉害的YOLOX来了，官方支持OpenVINO推理
本文参与 腾讯云自媒体同步曝光计划，分享自微信公众号。
原始发表：2022-04-15，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
用户4248041
妹子，精通CSS、JavaScript、PHP、ASP、 C、C＋＋、C#、Java、Ruby、Perl、 Lisp、python、Objective-C、 ActionScript、Pascal等单词的拼写，搞的我还以为是精通CSS、JavaScript、PHP、ASP、 C、C＋＋、C#、Java、Ruby、Perl、 Lisp、python、Objective-C、 ActionScript、Pascal，你要是有这么多精通，我不可能没听说过你！
妹子，精通CSS、JavaScript、PHP、ASP、 C、C＋＋、C#、Java、Ruby、Perl、 Lisp、python、Objective-C、 ActionScript、Pascal等单词的拼写，搞的我还以为是精通CSS、JavaScript、PHP、ASP、 C、C＋＋、C#、Java、Ruby、Perl、 Lisp、python、Objective-C、 ActionScript、Pascal，你要是有这么多精通，我不可能没听说过你！
回复回复点赞举报
推荐阅读
相关推荐
40个只有程序员才看得懂的段子
更多 >
腾讯云开发者
