YOLOv5 一键安装器:Windows / Ubuntu / WSL 下快速部署 CPU/GPU 环境
·
YOLOv5 一键安装器:Windows / Ubuntu / WSL 下快速部署 CPU/GPU 环境
前言
一个面向 Windows / Ubuntu / WSL 的 YOLOv5 一键安装器,帮助用户快速完成 CPU / GPU 环境部署、项目同步、依赖安装、权重补全与启动脚本生成。
平时在部署 YOLOv5 时,比较麻烦的往往不是模型本身,而是环境准备过程:
Python版本要对PyTorch要区分CPU / GPUConda、依赖、项目文件、预训练权重都要分别处理WSL下 GPU 运行时还经常需要额外折腾
所以我做了一个 YOLOv5 一键安装器,目标很简单:尽量把环境部署这件事做成开箱即用。
项目简介
这个安装器主要面向希望快速部署 YOLOv5 的用户,支持在不同平台下自动完成以下内容:
- 安装或复用
Conda - 同步
YOLOv5项目文件 - 创建虚拟环境
- 安装基础依赖
- 安装
CPU / GPU版PyTorch - 自动执行环境验证与推理验证
- 生成一键启动脚本
对于不想手动处理环境细节的用户来说,会省很多时间。
支持平台
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间切换的人 - 想快速验证
CPU或GPU是否能跑通的人 - 希望把安装流程交给脚本统一处理的人
安装方式
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 的环境准备折腾,尤其是在 Windows、Ubuntu、WSL 之间来回切换,那么这种“尽量自动化”的安装器,还是很省时间的。
它不追求花哨,核心目标就是两点:
- 少走弯路
- 尽快跑起来
更多推荐




所有评论(0)