项目预览
项目数据
项目介绍
Star 增长趋势
本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整
数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 5194
使用教程
ℹ 本教程由 AI 依据该项目 README 整理,个别步骤可能需要按你的环境微调。
环境要求
- Ubuntu 16.04+ 操作系统
- 至少一块较新的 NVIDIA GPU,驱动版本 >= 470.57.02
- 较大模型建议 GPU 显存 24GB 以上
- Python 3.10(README 推荐用 uv 管理环境)
- Docker 方式需先安装 NVIDIA Container Toolkit
安装与启动步骤
-
1确认硬件与系统
官方要求 Ubuntu 16.04+ 与至少一块较新 NVIDIA GPU,驱动不低于 470.57.02;跑大模型建议显存 24GB 以上。
-
2安装 NVIDIA 驱动
裸机跑 Ubuntu 时需自行装驱动和 CUDA,以下命令以 Ubuntu 22.04 为例,可按自己系统调整。
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-ubuntu2204.pin sudo mv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600 wget https://developer.download.nvidia.com/compute/cuda/12.4.0/local_installers/cuda-repo-ubuntu2204-12-4-local_12.4.0-550.54.14-1_amd64.deb sudo dpkg -i cuda-repo-ubuntu2204-12-4-local_12.4.0-550.54.14-1_amd64.deb sudo cp /var/cuda-repo-ubuntu2204-12-4-local/cuda-*-keyring.gpg /usr/share/keyrings/ sudo apt-get update sudo apt-get -y install cuda-toolkit-12-4 -
3准备运行环境
官方推荐用 uv 搭配 Python 3.10 安装,README 未给出具体安装命令,请按官方 Set up 文档操作。
-
4启动图形界面
在项目目录执行该命令会启动 H2O Wave 服务和应用,随后可打开浏览器访问。
make llmstudio -
5Docker 方式启动
已装 NVIDIA Container Toolkit 时可用官方镜像启动,挂载本地目录并映射 10101 端口。
mkdir -p `pwd`/llmstudio_mnt chmod 777 `pwd`/llmstudio_mnt docker run \ --runtime=nvidia \ --shm-size=64g \ --init \ --rm \ -it \ -u `id -u`:`id -g` \ -p 10101:10101 \ -v `pwd`/llmstudio_mnt:/mount \ h2oairelease/h2oai-llmstudio-app:latest -
6命令行微调模型
用 CLI 指定包含全部实验参数的 yaml 配置文件训练,把路径换成你自己的配置文件。
uv run python llm_studio/train.py -Y /your/path/config.yaml -
7多卡 DDP 训练
以 DDP 模式多卡训练,把数字换成 GPU 数量;默认使用前 k 块 GPU,可用 CUDA_VISIBLE_DEVICES 指定。
bash distributed_train.sh 2 -Y /your/path/config.yaml -
8与模型对话
启动交互式对话,experiment_name 为实验输出目录,UI 微调的模型同样可用。
uv run python llm_studio/prompt.py -e my_experiment
关键配置
| 配置项 | 必填 | 说明 | 示例 |
|---|---|---|---|
config_yaml | 是 | CLI 训练所需的实验参数配置文件路径 | /your/path/config.yaml |
CUDA_VISIBLE_DEVICES | 否 | 指定训练实际使用的 GPU 编号 | 0,1 |
H2O_WAVE_MAX_REQUEST_SIZE | 否 | 非 uv 环境启动 GUI 时需设置的最大请求体积 | 25MB |
H2O_WAVE_NO_LOG | 否 | 非 uv 环境启动 GUI 时关闭日志输出 | true |
H2O_WAVE_PRIVATE_DIR | 否 | 非 uv 环境启动 GUI 时的私有目录 | /download/@output/download |
safe_serialization | 否 | 发布到 Hugging Face 时是否启用安全序列化,默认 True | True |
如何确认成功
浏览器(推荐 Chrome)打开 http://localhost:10101/ ,能看到 H2O LLM Studio 界面即启动成功。
常见问题
Q:没有 NVIDIA GPU 能用吗?
A:官方要求 Ubuntu 16.04+ 和至少一块较新的 NVIDIA GPU(驱动 >= 470.57.02),大模型建议 24GB 以上显存,纯 CPU 环境未在 README 中说明支持。
Q:用非 uv 环境如何启动 GUI?
A:设置 H2O_WAVE_MAX_REQUEST_SIZE=25MB、H2O_WAVE_NO_LOG=true、H2O_WAVE_PRIVATE_DIR="/download/@output/download" 后执行 wave run llm_studio.app。
Q:Docker 容器怎么管理?
A:可用 docker ps 查看运行中的容器,用 docker kill 停止容器,README 中给出了这两个常用命令。
Q:如何把模型发布到 Hugging Face?
A:使用 uv run python llm_studio/publish_to_hugging_face.py,传入实验目录、设备和 api_key 等参数,未登录时 user_id 和 api_key 可省略。
注意事项
- README 未给出 uv 与 Python 3.10 的具体安装命令,需参考官方 Set up 文档。
- GUI 默认端口为 10101,浏览器请访问 http://localhost:10101/。
- CLI 训练必须自备包含全部实验参数的 yaml 配置文件。
- Docker 方式需提前安装 nvidia-container-toolkit。
核心亮点
- 提供完整的无代码 GUI,降低微调门槛
- 内置超参数调优和早停机制,提升训练效率
- 支持多种主流 LLM 架构,兼容性强
不足之处
- 文档/社区待观察
- 对大规模分布式训练的支持可能不如原生框架灵活
适用场景
- 企业快速构建领域专属聊天机器人
- 研究人员实验不同微调策略
- 教学演示和初学者学习 LLM 微调
替代项目
Hugging Face Transformers、LlamaIndex、Axolotl
同类项目推荐
Scaling-Diffusion-Transformers-m···
开源
让扩散模型从小规模到大规模训练,超参数无需重调,省时省力。
[NeurIPS 2025] Official implementation for our paper "Scaling Diffusion Transformers···
sagemaker-training-toolkit
开源
把训练脚本塞进容器,SageMaker 自动跑训练并上传模型
Train machine learning models within a Docker container using Amazon SageMaker.
Sklearn-genetic-opt
开源
用遗传算法自动调参,告别网格搜索的低效
Hyperparameter optimization and feature selection for scikit-learn using evolutionar···
mlx-tune
开源
在 Mac 上本地微调大模型,多模态全支持,速度飞快
Fine-tune LLMs on your Mac with Apple Silicon. SFT, DPO, GRPO, Vision, TTS, STT, Emb···
self-llm
开源
手把手教你微调部署开源大模型,中文教程一步到位
《开源大模型食用指南》针对中国宝宝量身打造的基于Linux环境快速微调(全参数/Lora)、部···
distill-sd
开源
把大模型蒸馏成小模型,低显存也能跑高质量图像生成
Segmind Distilled diffusion
redco
开源
一键自动化分布式训练推理,轻松搞定联邦学习与隐私保护
NAACL '24 (Best Demo Paper RunnerUp) / MlSys @ NeurIPS '23 - RedCoast: A Lightweight···
MLX-LoRA-Studio
开源
Mac 上点几下就完成大模型 LoRA 微调,数据不出本机
A native Mac App for LLM fine-tuning on Apple Silicon — fully on-device, fully open···
WaDi
开源
一步生成高质量图像,蒸馏时精准把控权重方向
[CVPR 2026] Official PyTorch implementation of WaDi: Weight Direction-aware Distilla···
