Apple Silicon Mac で GPU アクセラレーションによるトレーニングを可能にする機械学習フレームワーク PyTorch

Apple の Metal エンジニアリング チームと協力して、PyTorch は本日、同社のオープンソース機械学習フレームワーク が 、M1、M1 Pro、M1 Max、または M1 Ultra チップを搭載した Apple シリコン Mac での GPU アクセラレーションによるモデル トレーニングを間もなくサポートすることを発表しました。

これまで、Mac 上での PyTorch トレーニングでは CPU のみが利用されていましたが、次期バージョンでは、開発者や研究者が Apple シリコン チップに統合された GPU を利用して、モデルのトレーニングを「大幅に高速化」できるようになります。

GPU アクセラレーション トレーニングを備えた PyTorch バージョン 1.12 のプレビュー ビルドは、Python のネイティブ バージョンを備えた macOS 12.3 以降を実行している Apple Silicon Mac で利用できます。パフォーマンスの比較と追加の詳細は 、PyTorch の Web サイトで参照できます 。