PyTorch on Apple Silicon
18 05 22 - Filed in: Apple silicon M1 chip Machine learning
PyTorch is now available on Apple Silicon https://pytorch.org/blog/introducing-accelerated-pytorch-training-on-mac/.
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 learning workflows like prototyping and fine-tuning locally, right on Mac.
To get started, just install the latest Preview (Nightly) build on your Apple silicon Mac running macOS 12.3 or later with a native version (arm64) of Python https://pytorch.org/get-started/locally/
blog comments powered by Disqus