Lightning-AI/pytorch-lightning

これは何？

主な機能

• 01Trainerにより、同じコードを単一のCPUまたはGPUから複数ノードのクラスターまでスケールできます。
• 02TPU、混合精度、DDP・FSDP・DeepSpeedなどの分散戦略に対応しています。
• 03Lightning Fabricを使うと、独自の学習ループを細かく制御できます。
• 04チェックポイント保存、早期終了、ロギング連携を標準で利用できます。
• 05モデルをTorchScriptおよびONNXにエクスポートできます。

ユースケース

• 定型的な学習コードを書かずに、PyTorchで深層学習モデルを学習またはファインチューニングする。
• 既存のPyTorch学習スクリプトを、マルチGPUまたはTPU環境で実行できるようにする。
• 分散処理や混合精度のプリミティブを活用して、独自の学習ループを構築する。
• 実験追跡機能を統合し、再現性のある実験を実行する。

クイックスタート

1. 1

   pipでLightningをインストールします

   pip install lightning
2. 2

   サンプル用にtorchvisionをインストールします

   pip install torchvision
3. 3

   簡単なサンプルをmain.pyとして保存し、実行します

   python main.py

収録されている特集

トピック

よくある質問

pytorch-lightning とは？

モデルコードの完全な制御を保ちながら、学習ループやマルチGPU対応、定型的なエンジニアリング作業を自動化するPyTorchベースの深層学習フレームワーク。 PyTorch Lightningは、PyTorchモデルの事前学習とファインチューニング向けのフレームワークです。LightningModuleにモデルのロジックを定義すると、Trainerが誤差逆伝播、混合精度、デバイスの割り当て、CPU・GPU・TPU・複数ノードにまたがる分散学習を処理します。Trainerの引数を少し変更するだけで、同じコードを異なるハードウェア上で実行できます。

pytorch-lightning は無料で使えますか？ライセンスは？

pytorch-lightning は Apache-2.0 ライセンスのオープンソースです。商用製品で使う前にライセンス条件を確認してください。

pytorch-lightning は今もメンテナンスされていますか？

活発にメンテナンス中。pytorch-lightning の最終コミットは 2026年10月30日 です。

pytorch-lightning は何に使えますか？

定型的な学習コードを書かずに、PyTorchで深層学習モデルを学習またはファインチューニングする。 既存のPyTorch学習スクリプトを、マルチGPUまたはTPU環境で実行できるようにする。 分散処理や混合精度のプリミティブを活用して、独自の学習ループを構築する。 実験追跡機能を統合し、再現性のある実験を実行する。

pytorch-lightning のインストール方法は？

pipでLightningをインストールします: pip install lightning サンプル用にtorchvisionをインストールします: pip install torchvision 簡単なサンプルをmain.pyとして保存し、実行します: python main.py

pytorch-lightning はどこで入手できますか？

ソースコードとインストール手順は GitHub にあります：https://github.com/Lightning-AI/pytorch-lightning