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