PyTorch、Mac（Apple Silicon M1）でGPUアクセラレーションをサポート

• AppleのMetal開発チームとの協業により、PyTorchがMacでGPUアクセラレーションを利用可能に
  • MPS（Metal Performance Shaders）をPyTorchのバックエンドとして活用
• VGG16でCPUを使用した場合と比べて、Trainingは8倍、Evaluationは21倍の性能向上を示す

[ninebow]

PyTorch韓国ユーザー会の投稿もそっと添えておきます ^^;

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

[ninebow]

Apple M1 デバイスで PyTorch 1.12 をインストールし、MPS デバイスを使用する方法をまとめました。

conda は arm64 をサポートしていないため利用が難しい場合がありますが、対応状況の確認や解決方法などもあわせて整理しました。

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