
PaddleFormers
在飞桨上跑通大模型微调与训练,开箱即用
项目预览
项目数据
项目介绍
Star 增长趋势
本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整
数据来源:GitHub API 逐日快照 · 收录于 2026-09-15 · 当前星标 ★ 12968
技术标签
使用教程
环境要求
- Python ≥ 3.12(安装章节实际支持 python 3.10~3.13)
- CUDA ≥ 13.0(官方镜像同时提供 cuda12.6 / cuda12.9 / cuda13.0 版本)
- PaddleFleet ≥ 0.3(仅 GPU 训练功能需要)
- 操作系统:Linux 或 Windows
- 使用容器方式时需要 Docker 与 NVIDIA 显卡驱动(--gpus all)
安装与启动步骤
-
1Docker 方式(推荐)
用官方预置镜像起容器,镜像内已克隆 PaddleFormers 仓库并完成安装,避免本地环境冲突。按显卡选择 cuda12.6/12.9/13.0 镜像标签。
docker run --gpus all --name paddleformers-work -v $(pwd):/work \ -w=/work --shm-size=512G --network=host -it \ ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlepaddle/paddle:3.3.0-gpu-cuda12.6-cudnn9.5 /bin/bash -
2创建虚拟环境
不使用 Docker 时,推荐先用 conda / venv / uv 隔离 Python 环境,再安装依赖。以下以 conda 为例。
conda create -n paddleformers-work python=3.10 conda activate paddleformers-work -
3源码安装
克隆仓库后以可编辑模式安装,并附带 paddlefleet 依赖;按实际 CUDA 版本替换命令中的 cu126 索引地址。
git clone https://github.com/PaddlePaddle/PaddleFormers.git cd PaddleFormers python -m pip install -e '.[paddlefleet]' --extra-index-url https://www.paddlepaddle.org.cn/packages/nightly/cu126/ --extra-index-url https://www.paddlepaddle.org.cn/packages/stable/cu126/ -
4pip 直接安装
不想拉取源码时,可直接从 pip 安装 PaddleFormers 与 PaddleFleet,同样需按 CUDA 版本选择索引地址。
python -m pip install "paddleformers[paddlefleet]" --extra-index-url https://www.paddlepaddle.org.cn/packages/stable/cu126/ -
5轻量安装 tokenizer
如果只使用 tokenizer 或 processor,可只装基础包,不安装训练相关依赖,安装速度更快。
python -m pip install paddleformers
如何确认成功
安装或 docker run 命令无报错,进入容器/虚拟环境后执行 python -c "import paddleformers" 能正常导入即成功。
常见问题
Q:环境依赖写 Python ≥ 3.12,但示例用 python=3.10,到底用哪个?
A:安装章节说明支持 python 3.10~3.13,示例给出的就是 3.10。建议在该区间内选择,若追求与依赖说明一致可用 3.12 及以上版本。
Q:不同 CUDA 版本怎么安装?
A:替换命令中的索引地址与镜像标签即可:cuda12.6 用 cu126,cuda12.9 用 cu129,cuda13.0 用 cu130,三个版本 README 均已给出。
Q:只用 tokenizer 也要装 PaddleFleet 吗?
A:不需要。PaddleFleet 仅为 GPU 训练功能依赖,直接执行 python -m pip install paddleformers 即可,不会安装训练相关依赖。
Q:昆仑芯、天数智芯、沐曦、摩尔线程等其他硬件怎么装?
A:README 提供了独立的安装说明文档,分别在 docs/zh/XPU_installation_guide.md、ILUVATAR-GPU_installation_guide.md、Metax-GPU_installation_guide.md、MUSA-GPU_installation_guide.md。
注意事项
- Docker 方式被 README 标注为推荐,容器已拉取仓库并完成安装,可直接进入 /work 目录开始工作。
- docker run 使用 --shm-size=512G 与 --network=host,训练大模型时不要随意改小共享内存。
- 官方镜像标签为 paddle:3.3.0-gpu-cuda12.6-cudnn9.5,另有 cuda12.9-cudnn9.9 与 cuda13.0-cudnn9.13 可选。
- 命令示例中统一以 cuda12.6(cu126)为例,其他版本请按注释替换为 cu129 或 cu130。
核心亮点
- 深度绑定 PaddlePaddle,国产框架用户无需自行移植 PyTorch 训练代码
- 覆盖 Llama、Qwen、DeepSeek 等主流模型,权重与脚本开箱可用
- 支持 LoRA、全量微调及多种并行策略,兼顾单卡与大规模分布式训练
不足之处
- 生态与社区规模远小于 Hugging Face Transformers,第三方扩展和教程较少
- 强依赖 PaddlePaddle,非飞桨用户迁移成本高,跨框架复用性弱
适用场景
- 国内团队在国产算力与飞桨栈上微调开源大模型
- 企业基于 PaddlePaddle 做领域大模型的继续预训练
- 教学或科研中用统一脚本对比不同开源模型的训练效果
替代项目
Hugging Face Transformers、LLaMA-Factory、Megatron-LM
下一篇:Skills
同类项目推荐
WaDi
开源
一步生成高质量图像,蒸馏时精准把控权重方向
[CVPR 2026] Official PyTorch implementation of WaDi: Weight Direction-aware Distilla···
Dreambooth-Stable-Diffusion
开源
用几张照片,训练专属 Stable Diffusion 模型,生成你的定制图像。
Implementation of Dreambooth (https://arxiv.org/abs/2208.12242) by way of Textual In···
brewery-ai
开源
零经验也能在控制台微调 AI 模型
An intuitive console agent to fine-tune AI models with little to no experience requi···
ReaLHF
开源
重新分配参数,让 RLHF 训练更快更省资源
Super-Efficient RLHF Training of LLMs with Parameter Reallocation
dFactory
开源
几行代码微调扩散语言模型,省显存快验证
Easy and Efficient dLLM Fine-Tuning
data-forcing-distillation
开源
少步生成视频,不丢多样性和画质
Data-Forcing Distillation (DFD): restoring diversity and fidelity in few-step video ···
petals
开源
在家用 BitTorrent 方式跑大模型,推理微调快 10 倍
Run LLMs at home, BitTorrent-style. Fine-tuning and inference up to 10x faster than ···
DeepGNN
开源
让十亿级图数据的 GNN 训练跑得动、跑得快
DeepGNN is a framework for training machine learning models on large scale graph dat···
Fast-LLM
开源
让 LLM 训练跑满 GPU,少等快出结果
Accelerating your LLM training to full speed! Made with ❤️ by ServiceNow Research