项目预览
项目数据
项目介绍
Star 增长趋势
本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整
数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 12674
技术标签
使用教程
ℹ 本教程由 AI 依据该项目 README 整理,个别步骤可能需要按你的环境微调。
环境要求
- NVIDIA GPU(Ampere 或更新架构,用于 bf16 与 Flash Attention)或 AMD GPU
- Python >= 3.11(推荐 3.12)
- PyTorch >= 2.11.0(推荐 2.12.1)
- 本机已安装 Docker(使用 Docker 方式时)
安装与启动步骤
-
1确认硬件与版本
先检查显卡是否为 NVIDIA Ampere 或更新架构,或 AMD GPU;同时确认 Python 与 PyTorch 版本满足最低要求。
-
2安装 uv 包管理器
README 给出的安装前置动作:执行脚本安装 uv,如果本机已有 uv 可跳过。
curl -LsSf https://astral.sh/uv/install.sh | sh -
3重启终端
uv 安装完成后需要重启 shell(重新打开终端),让 uv 命令进入 PATH 后再生效。
-
4用 Docker 启动 axolotl
README 说明 Docker 安装比在自有环境安装更不易出错,该命令挂载全部 GPU 并以交互方式进入容器。
docker run --gpus '"all"' --ipc=host --rm -it axolotlai/axolotl:main-latest -
5查看其他安装方式
若不想用 Docker,README 指向官方文档的安装页面,其中列出了其余安装方式。
如何确认成功
README 未给出验证命令;容器启动后可参考官方文档 docs.axolotl.ai 确认环境是否可用。
常见问题
Q:Docker 安装和本地安装该选哪个?
A:README 明确说明:用 Docker 安装比在自己环境中安装更不容易出错,新手建议先用 Docker 方式。
Q:为什么要求 Ampere 或更新的 NVIDIA GPU?
A:README 指出 bf16 和 Flash Attention 需要 Ampere 或更新架构的 NVIDIA GPU;不满足时可用 AMD GPU。
Q:Python 和 PyTorch 版本有硬性要求吗?
A:README 要求 Python >= 3.11(推荐 3.12),PyTorch >= 2.11.0(推荐 2.12.1)。
Q:安装 uv 后提示找不到 uv 命令?
A:README 提示安装脚本执行后需要重启 shell(重新打开终端)才能生效。
Q:还有哪些安装方式?
A:README 指向官方文档安装页:https://docs.axolotl.ai/docs/installation.html。
注意事项
- Docker 命令中的 --gpus '"all"' 表示暴露全部 GPU,--ipc=host 用于容器与主机共享内存。
- uv 安装脚本执行后必须重启终端,否则命令不可用。
- README 节选未提供本地 pip/源码安装的具体命令,请以官方文档安装页为准。
- axolotl 是命令行微调框架,具体训练需自行准备 YAML 配置,README 节选未给出示例配置。
核心亮点
- 统一 YAML 配置,无需编写复杂训练脚本
- 内置 LoRA/QLoRA、Flash Attention 等优化,显存占用低
- 支持主流模型架构和数据集格式,开箱即用
不足之处
- 文档偏重使用示例,对内部机制解释较少
- 依赖较多,环境配置可能稍显繁琐
适用场景
- 基于开源大模型做领域微调(如法律、医疗)
- 在有限 GPU 资源下进行参数高效微调
- 批量实验不同模型和超参,快速对比效果
替代项目
Hugging Face TRL、LLaMA-Factory、Unsloth
同类项目推荐
minAlphaFold2
开源
用 PyTorch 极简复现 AlphaFold2,读懂蛋白质结构预测
A minimal PyTorch re-implementation of AlphaFold2's model & training
DS2
开源
用 LLM 评分自动筛数据,少数据也能训出好模型
[ICLR 2025] Official implementation of paper "Improving Data Efficiency via Curating···
BitVAE
开源
用比特级分词,让图像生成更精细高效
official training and inference code of bitwise tokenizer
meta-context-engineering
开源
让AI自动进化上下文策略,告别手写提示词
[ICML 2026] Meta Context Engineering via Agentic Skill Evolution
model-optimization
开源
几行代码压缩模型,让大模型跑进手机和嵌入式设备
A toolkit to optimize ML models for deployment for Keras and TensorFlow, including q···
Open-LLaVA-NeXT
开源
开源复现 LLaVA-NeXT,快速训练你的多模态对话模型
An open-source implementation for training LLaVA-NeXT.
TFMQ-DM
开源
给扩散模型瘦身,4-bit量化不降画质,部署提速。
[CVPR 2024 Highlight & TPAMI 2025] This is the official PyTorch implementation of "T···
ReCall
开源
让大模型学会自己上网查资料,回答更准更及时
ReSearch: Learning to Reason with Search for LLMs via Reinforcement Learning & ReCal···
FineTuningLLMs
开源
跟着书一步步微调大模型,从零跑通全流程
Official repository of my book "A Hands-On Guide to Fine-Tuning LLMs with PyTorch an···
