PyTorch, 맥에서 GPU 가속 지원

PyTorch는 애플의 Metal 엔지니어링 팀과 협업을 통해 개발한 Mac에서의 PyTorch의 GPU 가속 지원을 발표했습니다. 지금까지 Mac에서의 PyTorch 학습은 CPU만 사용했지만, 이번 v1.12 릴리즈를 통해 개발자와 연구자들은 Apple Silicon GPU를 활용해 훨씬 빠르게 모델을 학습할 수 있습니다. 이를 통해 Mac에서 프로토타이핑이나 Fine-tuning 등의 머신러닝 워크플로우를 바로 수행할 수 있습니다. 모든 Apple Silicon Mac은 Unified Memory 아키텍처를 갖고 있어 GPU에서 전체 메모리 영역에 직접 접근할 수 있습니다. 이러한 특징은 사용자가 더 큰 모델이나 더 큰 배치 크기로 로컬에서 학습을 수행할 수 있게 하며, Mac을 머신 러닝을 위한 훌륭한 플랫폼으로 만들어주어 로컬 GPU나 클라우드 환경 구축에 들어가는 비용을 절감할 수 있습니다. 또한 Unified Memory 아키텍처는 데이터 읽기 레이턴시를 감소시켜 End-to-End 성능을 향상시킵니다. PyTorch의 Mac GPU 가속 지원은 macOS 12.3 이후의 버전 및 네이티브 arm64 버전의 Python 을 구동하는 Apple Silicon Mac에서 프리뷰 빌드를 설치해 사용할 수 있습니다.