当前位置:   article > 正文

YOLOv8目标检测TensorRT加速_yolov8 tensorrt

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安装验证

  1. import tensorrt as trt
  2. # 尝试创建一个TensorR
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/我家小花儿/article/detail/1013285
推荐阅读
相关标签
  

闽ICP备14008679号