[yolov5目标检测] Ubuntu18.04安装tensorrt加速网络推理

最新推荐文章于 2024-07-19 11:20:37 发布

scut_lrr

最新推荐文章于 2024-07-19 11:20:37 发布

阅读量1.5k

点赞数 3

分类专栏： pytorch 深度学习环境搭建 yolo

本文链接：https://blog.csdn.net/weixin_42448226/article/details/109677174

版权

本文详细介绍了在Ubuntu18.04上安装CUDA10.0和cuDNN7.6.5，然后下载TensorRT的deb文件并安装。接着，克隆YOLOv5项目并按照readme.md进行检测和训练流程。通过使用TensorRT，将YOLOv5s模型转换为.wts文件，编译项目生成.engine文件，实现推理速度的显著提升。

摘要由CSDN通过智能技术生成

1.安装cuda10.0和cudnn7.6.5

参考：tensorrt环境搭建

dpkg -l | grep cuda

查看cuda安装情况，若发现libcudnn版本不适应（即libcudnn版本不是10.0）

使用以下命令删除：

sudo dpkg --purge libcudnn7-dev
sudo dpkg --purge libcudnn7

2.下载tensorrt的deb文件并安装

参考：nVidia官方安装指导

sudo dpkg -i nv-tensorrt-repo-${os}-${tag}_1-1_amd64.deb
sudo apt-key add /var/nv-tensorrt-repo-${tag}/7fa2af80.pub

sudo apt-get update
sudo apt-get install tensorrt


example：
sudo dpkg -i nv-tensorrt-repo-ubuntu1804-cuda10.0-trt7.0.0.11-ga-20191216_1-1_amd64.deb
sudo apt-key add /var/nv-tensorrt-repo-ubuntu1804-cuda10.0-trt7.0.0.11-ga-201912