반응형 DEEPLEARNING1 [딥러닝] 파이토치 multi GPU 로 학습시키기 (feat. Pytorch lightning ) 딥러닝에서 여러개의 gpu를 이용해서 학습을 시키면 학습에 걸리는 시간을 줄일 수 있습니다. Pytorch를 이용해서 코딩을 해도 되고 Pytorch lightning을 이용해서 보다 간편하게 학습을 시킬 수도 있습니다. 본 글에서는 파이토치 라이트닝을 이용해서 Multi-GPU로 학습시킬 때 주의할 점에 대해서 말씀드리겠습니다. 병렬학습이라고도 할 수 있습니다. 확인사항우선 시스템에서 GPU를 잘 인지하는지 확인해봅시다. 만약 에러가 날 경우에는 많은 경우 올바른 버전의 패키지를 설치하지 않았기 때문입니다. 우선 쥬피터 노트북이든 그냥 파이썬이든 열어서 아래와 같은 명령어를 입력해봅시다. import torch, os#보통 이 두 명령어면 됩니다.#GPU 인지 되는지 확인print(torch.cuda... 2025. 1. 14. 이전 1 다음 반응형