YOLOv5 一键安装器:Windows / Ubuntu / WSL 下快速部署 CPU/GPU 环境

前言

一个面向 Windows / Ubuntu / WSLYOLOv5 一键安装器,帮助用户快速完成 CPU / GPU 环境部署、项目同步、依赖安装、权重补全与启动脚本生成。

平时在部署 YOLOv5 时,比较麻烦的往往不是模型本身,而是环境准备过程:

  • Python 版本要对
  • PyTorch 要区分 CPU / GPU
  • Conda、依赖、项目文件、预训练权重都要分别处理
  • WSL 下 GPU 运行时还经常需要额外折腾

所以我做了一个 YOLOv5 一键安装器,目标很简单:尽量把环境部署这件事做成开箱即用。


项目简介

这个安装器主要面向希望快速部署 YOLOv5 的用户,支持在不同平台下自动完成以下内容:

  • 安装或复用 Conda
  • 同步 YOLOv5 项目文件
  • 创建虚拟环境
  • 安装基础依赖
  • 安装 CPU / GPUPyTorch
  • 自动执行环境验证与推理验证
  • 生成一键启动脚本

对于不想手动处理环境细节的用户来说,会省很多时间。


支持平台

Windows 安装器

支持平台:Windows x64,支持 CPU / NVIDIA GPU(CUDA 版本 >= 11.6) 环境

Ubuntu 安装器

支持平台:Ubuntu x64 / WSL(Ubuntu),支持 CPU / NVIDIA GPU(CUDA 版本 >= 11.6) 环境


目前支持的内容

  • 支持 YOLOv5 v5.0 / v6.1 / v7.0
  • 支持 CPU 安装
  • 支持 NVIDIA GPU 安装
  • 支持 智能安装
  • 支持补全对应版本的预训练权重
  • 支持生成桌面/终端一键启动脚本
  • 支持卸载、深度清理、环境修复

这个安装器适合谁

我觉得比较适合下面几类用户:

  • 刚接触 YOLOv5,不想卡在环境阶段的人
  • 需要频繁在 Windows / Ubuntu / WSL 间切换的人
  • 想快速验证 CPUGPU 是否能跑通的人
  • 希望把安装流程交给脚本统一处理的人

安装方式

Windows

可以直接运行:

powershell -NoProfile -ExecutionPolicy Bypass -Command "irm 'https://yolo.bmclx.top/install.ps1' | iex"

Ubuntu / WSL(Ubuntu)

bash <(curl -fsSL https://yolo.bmclx.top/install.sh)

在这里插入图片描述

使用体验上的几个点

这个安装器目前做得比较实用的地方,主要有这些:

  • 会根据环境自动判断更适合走 CPU 还是 GPU
  • WSL 下会自动处理 GPU 运行时相关路径
  • 安装完成后会自动做推理验证
  • 智能安装会默认选择当前推荐项目版本,并自动完成环境校验、推理验证和启动脚本生成
  • 补全权重时支持按数字序号单选、多选、全部下载
  • 权重补全会严格按当前项目版本处理,不会跨版本混用

结语

如果你也经常被 YOLOv5 的环境准备折腾,尤其是在 WindowsUbuntuWSL 之间来回切换,那么这种“尽量自动化”的安装器,还是很省时间的。

它不追求花哨,核心目标就是两点:

  • 少走弯路
  • 尽快跑起来
Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐