이제 Apple M1에서도 PyTorch GPU 가속이 가능해집니다! 🎉

[9bow (박정환)]

많은 분들께서 기다리고 기다리셨던,
Apple M1 칩에서의 GPU 가속 기능이 드디어, PyTorch 1.12부터 가능해진다고 합니다!

기존의 cuda 장치처럼 mps(Apple의 Metal Performance Shaders) 장치로 접근해서 사용할 수 있다고 합니다.
(아래 코드는 MPS Backend 문서에서 가져왔습니다.)

mps_device = torch.device("mps")

# Create a Tensor directly on the mps device
x = torch.ones(5, device=mps_device)
# Or
x = torch.ones(5, device="mps")

# Any operation happens on the GPU
y = x * 2

# Move your model to mps just like any other device
model = YourFavoriteNet()
model.to(mps_device)

# Now every call runs on the GPU
pred = model(x)

아래와 같이 M1 Ultra에서 학습과 평가 모두 CPU 대비 월등한 성능 향상을 보여주는 것을 보실 수 있습니다.

자세한 공지는 아래에서 보실 수 있으며, Preview 버전(Nightly Build)에서는 바로 사용할 수 있다고 합니다.

Introducing Accelerated PyTorch Training on Mac

기존의 많은 사용자들의 요청과 아우성(…)은 아래 이슈에서 보실 수 있습니다

GPU acceleration for Apple's M1 chip?

[9bow (박정환)]

다른 분들께도 필요할 것 같아서, 자주 묻는 질문&답변 게시판에 정리해서 올렸습니다.

내용은 거의 같지만, 필요하신 분들이 계실 것 같아 아래 링크를 남깁니다.

Apple M1/M2이 탑재된 장치에서 GPU 가속을 사용하려면 어떻게 해야 하나요?

공식 홈페이지와 StackOverflow 등에서 자주 보이는 질문과 답변을 번역하고 있습니다. 몇 일 전 소식이 전해진, 많은 분들께서 오랜 시간 기다리셨고 또 궁금해하시는 Apple M1 칩이 탑재된 맥북 또는 맥미니, 맥스튜디오 등에서 GPU 가속 사용 방법을 정리해보려고 합니다. 질문 Apple M1이 탑재된 장치에서 GPU 가속을 사용하려면 어떻게 해야 하나요? (또는, M1 칩이 탑재된 맥북 / 맥미니 / 맥스튜디오에서 PyTorch GPU 가속 기능을 하고 싶습니다.)