Introducing Accelerated PyTorch Training on Mac – PyTorch

In collaboration with the Metal engineering team at Apple, we are excited to announce support for GPU-accelerated PyTorch training on Mac. Until now, PyTorch training on Mac only leveraged the CPU, but with the upcoming PyTorch v1.12 release, developers and researchers can take advantage of Apple silicon GPUs for significantly faster model training. This unlocks the ability to perform machine lear

[tiryoh]

PyTorchのNightly buildでM1対応きてた👀

[Nyoho]

PyTorch 1.12のプレビューApple Siliconでかなり高速の模様。 device("mps") で利用できる。

[hnishi2509]

“In collaboration with the Metal engineering team at Apple, we are excited to announce support for GPU-accelerated PyTorch training on Mac.”

[misshiki]

“MacでのGPUで高速化されたPyTorchトレーニングのサポートを発表”