当前位置: 代码网 > it编程>前端脚本>Python > YOLOv8目标检测TensorRT加速

YOLOv8目标检测TensorRT加速

2024年08月03日 Python 我要评论
YOLOv8的TensorRT加速,模型导出和接口开发,接口调用。本博客仅当学习使用,切勿商用,谢谢!!!

一、tensorrt环境安装

经过多次踩坑,无论你是windows还是linux,或者ubuntu系统,下面的方法都可以完美安装。

1.1 官网下载

去官网下载,自己系统相对应的压缩包,这个一定要与自己的硬件环境相适配。

链接:nvidia tensorrt 8.x download | nvidia developer

我测试的是8.5 ga的版本,下载需要邮箱登录,自行注册就好。

注意:如果你的是ubuntu系统也是下载linux的就可

1.2 tensorrt安装

若使用的是anaconda,直接切换到自己的环境,查看python的版本。

python -v

解压下载的压缩包,进入到python文件夹里

找到对应的python版本,cp后面是python的版本(38就是python3.8), 再运行

pip install tensorrt-8.5.1.7-cp38-none-win_amd64.whl

将tensorrt加入到环境变量中, 系统变量path

1.3 tensorrt安装验证

import tensorrt as trt
 
# 尝试创建一个tensorrt的logger
logger 
(0)

相关文章:

版权声明:本文内容由互联网用户贡献,该文观点仅代表作者本人。本站仅提供信息存储服务,不拥有所有权,不承担相关法律责任。 如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 2386932994@qq.com 举报,一经查实将立刻删除。

发表评论

验证码:
Copyright © 2017-2025  代码网 保留所有权利. 粤ICP备2024248653号
站长QQ:2386932994 | 联系邮箱:2386932994@qq.com