一、安装

1 linux+gtx3080 

1.7.1修复了不认3系列显卡的bug。

anaconda3虚环境下：

pip3 install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html --trusted-host mirrors.aliyun.com

cd  /home/XXX/dev/yolo/yolov5-3.1

pip3 install -r requirements.txt --trusted-host mirrors.aliyun.com

1.7不行

3080sm_86  pytorch 1.7不认。 要装nightly 1.8才认。

但是yolov5 目前只支持到pytorch1.7。在pytorch1.8下 最简单的coco128 train 都报错。

2 win10+ gtx1060

用yolov5 官网https://blog.csdn.net/weixin_44813425/article/details/107944100

python train.py --img 480 --batch 16 --epochs 5 --workers 0 --weights yolov5s.pt --data warship.yaml