인공지능공부/프레임워크
pytorch lightning multi-gpu사용하기
pytorch lightning에 대한 전반적인 내용은 아래 포스팅을 참고해주세요. pytorch lightning을 사용해보자 pytorch는 가장 널리 쓰이는 딥러닝 프레임워크중 하나이죠. 딥러닝, 머신러닝을 공부하시는 분이라면 들어보셨을겁니다. 써보신 분들은 아시겠지만 pytorch 자체가 굉장히 자유도가 높은 프레임워 jaeyoon-95.tistory.com pytorch lightning은 multi-gpu또한 쉽게 사용하실 수 있습니다. 방법을 찾아보면 아래와 같이 trainer 선언 부에 gpu의 개수를 적어주면 되는데요. trainer = pl.Trainer(max_epochs=10,gpus=4,num_sanity_val_steps=1) 저도 위와 같이 적어주어 GPU 4개인 환경에서 돌리..
