12P by xguru 1달전 | favorite | 댓글 2개
  • 애플의 Metal 개발팀과 협업으로 PyTorch가 맥에서 GPU 가속 가능해짐
    • MPS(Metal Performance Shader)를 PyTorch의 백엔드로 활용
  • VGG16 으로 CPU 사용했을 때와 비교시 Training은 8배, Evaluation은 21배의 성능 향상을 보임
ninebow 1달전  [-]

살포시 파이토치 한국 사용자 모임의 글도 덧붙여봅니다 ^^;

https://discuss.pytorch.kr/t/apple-m1-pytorch-gpu/276

답변달기
ninebow 1달전  [-]

Apple M1 기기에서 PyTorch 1.12를 설치하고 MPS 장치를 사용하는 방법을 정리하였습니다.

conda에서 arm64를 지원하지 않아 사용이 어려우실 수 있는데, 지원 여부 확인 및 해결 방법 등을 함께 정리하였습니다.

https://discuss.pytorch.kr/t/apple-m1-gpu/286

답변달기