linux下运行yolo,yolov5 安装尝试

一、安装

1 linux+gtx3080

放弃。

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

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

2 win10+ gtx1060

的安装语句

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 FilesPython38libsite-packagesorchlibcusolverMg64_10.dll" or one of its dependencies.

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

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