YoloV5训练自己的数据集(CPU与GPU)

Answer-six

已于 2025-01-08 12:55:36 修改

阅读量1.7k

点赞数 23

CC 4.0 BY-SA版权

文章标签： YOLO python 开发语言

于 2024-10-27 15:55:12 首次发布

本文链接：https://blog.csdn.net/yangjunsheng04/article/details/143082507

记录学习yolov5踩过的坑，yolov5的特点之一就是权重文件非常之小，可以搭载在配置更低的移动设备上.

写在开始配置之前-新手必看
CUDA
CUDA(Compute Unified Device Architecture，统一计算设备架构)，是显卡厂商NVIDIA在2007年推出的并行计算平台和编程模型。它利用图形处理器(GPU)能力，实现计算性能的显著提高。CUDA是一种由NVIDIA推出的通用并行计算架构，该架构使GPU能够解决复杂的计算问题，从而能通过程序控制底层的硬件进行计算。它包含了CUDA指令集架构(ISA)以及GPU内部的并行计算引擎。开发人员可以使用C/C++/C++11语言来为CUDA架构编写程序。CUDA提供host-device的编程模式以及非常多的接口函数和科学计算库，通过同时执行大量的线程而达到并行的目的。

Pytorch
Pytorch是torch的python版本，是由Facebook开源的神经网络框架，专门针对 GPU 加速的深度神经网络（DNN）编程。Torch 是一个经典的对多维矩阵数据进行操作的张量（tensor ）库，在机器学习和其他数学密集型应用有广泛应用。与Tensorflow的静态计算图不同，pytorch的计算图是动态的，可以根据计算需要实时改变计算图。但由于Torch语言采用 Lua，导致在国内一直很小众，并逐渐被支持 Python 的 Tensorflow 抢走用户。作为经典机器学习库 Torch 的端口，PyTorch 为 Python 语言使用者提供了舒适的写代码选择。

首先对于初学者来说最难的就是配置环境，在学习之前这里建议大家看看自己的电脑有没有独立显卡，用GPU跑模型比CPU快的非常多。配置环境在网上和B站有很多教程，慢慢摸索下来就知道了。

最重要的坑是：

1.torch，cuda，torchvision，python等等的对应关系，这个可以参考
原文链接：https://blog.csdn.net/crist_meng/article/details/136425444的教程，写的非常详细，python3.8我觉得非常好。

在配置完环境后需要使