ubuntu20.04搭建yolov5目标检测开发环境
Posted loongembedded
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了ubuntu20.04搭建yolov5目标检测开发环境相关的知识，希望对你有一定的参考价值。
文章目录
YOLOv5是完全基于Pytorch的，而Pytorch的主要性能发挥又依赖于CUDA和cuDNN，所以一定要安装好CUDA，cuDNN，和对应版本的Pytorch，这点非常重要，要是Pytorch和CUDA，cuDNN的版本不对应，YOLOv5的训练是跑不起来的。对应关系可以从Pytorch的官网： https://pytorch.org/ 上找到。
一、显卡驱动、CUDA和cudnn安装
以上是关于ubuntu20.04搭建yolov5目标检测开发环境的主要内容，如果未能解决你的问题，请参考以下文章
睿智的目标检测56——Pytorch搭建YoloV5目标检测平台
睿智的目标检测55——Keras搭建YoloV5目标检测平台