728x90
반응형
PyTorch Lightning은 딥러닝 프레임워크로, 이를 활용하면 누구나 쉽게 딥러닝 모델을 학습할 수 있습니다.
TensorFlow에 Keras가 있다면, PyTorch에는 PyTorch Lightning이 있다고 봐주시면 될 것 같습니다.
PyTorch Lightning은 아래 명령어로 간단하게 설치할 수 있습니다.
pip install pytorch-lightning
설치가 완료되면, 아래와 같이 import 해서 사용할 수 있습니다.
간단하게 버전 정보를 출력해 보도록 하겠습니다.
import pytorch_lightning as pl
print(f'PyTorch lightning version: {pl.__version__}')
위 출력 결과를 통해, PyTorch가 잘 설치되었는지, 어느 버전이 설치되었는지 확인할 수 있습니다.
728x90
반응형
'Deep Learning > PyTorch' 카테고리의 다른 글
| [PyTorch Lightning] Trainer 클래스에서 GPU 설정하는 방법 (2) | 2025.01.20 |
|---|---|
| [PyTorch] MLP 모델 만들기 2편 - 학습 모델 구축하기 (1) | 2025.01.13 |
| [Pytorch] DataLoader num_workers 개수 설정 방법 (3) | 2025.01.06 |
| [PyTorch] MLP 모델 만들기 1편 - 학습 데이터셋 만들기 (33) | 2025.01.01 |