yolov5 安装尝试

一、安装

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://github.com/ultralytics/yolov5

的安装语句

pip install -r requirements.txt

报

ERROR: Could not find a version that satisfies the requirement torch>=1.6.0 (from -r requirements.txt (line 12)) (from versions: 0.1.2, 0.1.2.post1, 0.1.2.post2)
ERROR: No matching distribution found for torch>=1.6.0 (from -r requirements.txt (line 12))

得去pytorch官网首页选自己的os cuda版本得到安装语句

pip install torch===1.7.0+cu110 torchvision===0.8.1+cu110 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

运行的话，下载torch 的whl有2G，要10多个小时，复制出地址：

https://download.pytorch.org/whl/cu110/torch-1.7.0%2Bcu110-cp38-cp38-win_amd64.whl

直接用迅雷下载搞定

然后继续运行这个语句，完成pytorch安装

然后继续执行

pip install -r requirements.txt

 

二、运行yolo v5

随便标注几张图像

 

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

报错

OSError: [WinError 1455] 页面文件太小，无法完成操作。 Error loading "C:\Program Files\Python38\lib\site-packages\torch\lib\cusolverMg64_10.dll" or one of its dependencies.

百度，发现需要在win下设置worker=0

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