zoukankan      html  css  js  c++  java
  • yolov5 安装尝试

    一、安装

    1 linux+gtx3080 

    放弃。

    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 FilesPython38libsite-packages orchlibcusolverMg64_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

     

  • 相关阅读:
    day55---前端基础之BOM操作和DOM操作
    每日作业5/8
    数据库之索引
    数据库之视图、触发器、事务、存储过程、内置函数、流程控制
    每日作业5/7
    数据备份与pymysql模块
    Navicat与MySQL使用
    每日作业5/6
    数据库之多表查询
    数据库之单表查询
  • 原文地址:https://www.cnblogs.com/xuanmanstein/p/14107844.html
Copyright © 2011-2022 走看看