PyTorch, 맥(애플실리콘 M1)에서 GPU 가속 지원
(pytorch.org)- 애플의 Metal 개발팀과 협업으로 PyTorch가 맥에서 GPU 가속 가능해짐
- MPS(Metal Performance Shader)를 PyTorch의 백엔드로 활용
- VGG16 으로 CPU 사용했을 때와 비교시 Training은 8배, Evaluation은 21배의 성능 향상을 보임
살포시 파이토치 한국 사용자 모임의 글도 덧붙여봅니다 ^^;
Apple M1 기기에서 PyTorch 1.12를 설치하고 MPS 장치를 사용하는 방법을 정리하였습니다.
conda에서 arm64를 지원하지 않아 사용이 어려우실 수 있는데, 지원 여부 확인 및 해결 방법 등을 함께 정리하였습니다.