win10下yolov5 v6.0环境配置

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
 
 
 
 

 

win10下yolov5 v6.0环境配置最先出现在Python成神之路。