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