PyTorch Lightning 模型训练框架
整理 PyTorch 训练代码,不改代码即可从单卡扩展到多卡多机训练
评分人数不足呦探分评分人数不足大众分评分方法
这个页面由 AI 根据官网公开信息整理,可能有错。产品团队可以认领后修改。 认领这个产品
这个项目只有源代码,需要会编程才能用。
- 星标
- 3.1万
- GitHub 下载量
- 12.7万
- 复刻
- 3843
- 主要语言
- Python
- 许可证
- Apache-2.0
- 最近更新
复制安装指令(给 AI 编程工具)
粘贴到 Codex、Claude Code、Cursor、Trae 等 AI 编程工具里,回车就会开始安装。第一次复制前请先阅读风险提示。
这些文件由项目作者放在 GitHub 上,本站不开发、不托管这些软件,也不检测其代码;国内访问 GitHub 可能较慢。使用前请阅读 第三方软件风险提示和项目的许可证。 数据更新于 2026/11/01 报告问题
介绍
PyTorch Lightning 是 Lightning AI 维护的深度学习框架,用于预训练和微调模型。它把 PyTorch 代码中的工程部分(反向传播、混合精度、多卡训练等)交给框架处理,使用者只需关注模型本身的逻辑。
能做什么
- 同一份代码可在 CPU、GPU、TPU 或多节点上运行,通过参数切换设备和精度
- 内置早停和检查点保存,接入 TensorBoard、MLflow 等日志工具,可导出 TorchScript、ONNX
- 用 Lightning Fabric 自行控制训练循环,支持 DDP、FSDP、DeepSpeed 等分布式策略
运行需要:Python 和 PyTorch,用 pip 或 conda 安装;GPU(CUDA 或 Apple 芯片)或 TPU 可选。
这段介绍由 AI 根据项目的公开说明整理,可能有疏漏,以项目主页为准。
评测
0 条评测还没有评测。用过的话,来写第一条吧。
写评测