1、首先安装cuda和cudnn，安装步骤参考
Cuda和cuDNN安装教程(超级详细)_jhsignal的博客-CSDN博客_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
版权声明：本文为CSDN博主「zhangfeng5909」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/zhangfeng5909/article/details/122243173
暂无评论