1、首先安装cuda和cudnn,安装步骤参考
Cuda和cuDNN安装教程(超级详细)_jhsignal的博客-优快云博客_cuda安装教程
2、安装PyTorch
pytorch:Start Locally | PyTorch

根据自己的cuda和cudnn选择对应的pytorch版本,如果是cuda10.1和10.2 LTS版本兼容性好点。
3、下载yolov5
https://github.com/ultralytics/yolov5/tree/v6.0
4、修改相应文件,自定义训练




5、自定义数据集


6、从头开始训练
python train.py --img 640 --batch 80 --epochs 800 --data data/coco128.yaml --cfg models/yolov5s.yaml
本文指导读者从安装CUDA和cuDNN,选择兼容PyTorch版本,到下载和修改Yolov5模型,自定义数据集,最后进行从头开始的训练全过程。适合深度学习爱好者和实践者。
959

被折叠的 条评论
为什么被折叠?



