PyTorch Lightning 模型训练框架

整理 PyTorch 训练代码,不改代码即可从单卡扩展到多卡多机训练

需要技术AI 整理,待开发者认领训练与微调
评分人数不足呦探分评分人数不足大众分评分方法
这个页面由 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 条评测

还没有评测。用过的话,来写第一条吧。

写评测