마지막으로 자세하게는 알아보지 않지만, 추가적으로 어떤 방법을 사용해서 Multi-GPU를 사용할 수 있는지 알아보는 글입니다. 추후에 사용할 일이 있다면 더 자세하게 작성할 것 같습니다.
Apex 엔비디아에서 제공하는 Multi-GPU 방법이라고 합니다. [ APEX ] 근데 살펴보니까 Apex에 작성된 코드들은 결국 Pytorch에 업스트림되고, 현재 Distributed leaning 관련한 코드들은 이미 파이토치에 업데이트 적용이 완료되었다고 합니다. 따라서 따로 만들어서 실험할 필요는 없는 것 같습니다.
또한 Apex를 사용하는 이유 중 하나인 AMP ( automatic mixed precision )도 곧 파이토치에 업로드가 전부 완료도리 것 같습니다. Pytorch Lightning 요즘 가장 핫합니다.
더럽게 모델을 안짜도되고, 편하게 GPU도 사용할 수 있습니다. 또한 Multi-GPU를 사용하는 방법도 다양해서 취향에 맞게 선택할 수 있습니다.
시간상 여기까지 직...
#
AI학습
#
spawn
#
torch
#
다중지피유
#
데이터분산학습
#
데이터패러럴
#
멀티쓰레드
#
멀티지피유
#
멀티프로세스
#
분산학습
#
인공지능GPU
#
Pytorch
#
MultiGPU
#
mpspawn
#
Apex
#
DataParallel
#
DDP
#
distributed
#
DistributedDataParallel
#
DP
#
GPU
#
GPU여러대사용하기
#
jjunsss
#
launch
#
파이토치
원문 링크 : Multi-GPU 총정리 5