PyTorch introduces GPU-accelerated training on Apple silicon Macs

A backend for PyTorch, Apple’s Metal Performance Shaders (MPS) help accelerate GPU training.
Listen to this story

PyTorch today announced a collaboration with Apple’s Metal engineering team to introduce support for GPU-accelerated PyTorch training on Mac systems powered by M1, M1 Pro, M1 Max and M1 Ultra chips. Up until now, PyTorch training on Macs was only for CPUs, but after the launch of PyTorch v1.12, developers can use Apple’s silicon GPUs to accelerate model training processes, like prototyping and fine-tuning on the Mac itself. 

Apple’s silicon Macs have a unified memory architecture that will provide GPUs with complete access to the full memory storage. A backend for PyTorch, Apple’s Metal Performance Shaders (MPS) help accelerate GPU training. MPS extends the PyTorch framework, offering scripts and frameworks for setting up and running operations on Macs. MPS optimises computing that is fine-tuned according to the varied features of each Metal GPU family. 

Performance speedup from accelerated GPU training compared to CPUs, Source: PyTorch

THE BELAMY

Sign up for your weekly dose of what's up in emerging technology.

Mac has become a reliable platform for training ML models. The unified memory architecture cuts down latency caused by data retrieval, reduces costs associated with cloud-based development and improves end-to-end performance.

A preview build of PyTorch v1.12 with GPU-accelerated training is available for macOS 12.3 or later with the native version of Python.


Download our Mobile App



More Great AIM Stories

Poulomi Chatterjee
Poulomi is a Technology Journalist with Analytics India Magazine. Her fascination with tech and eagerness to dive into new areas led her to the dynamic world of AI and data analytics.

AIM Upcoming Events

Early Bird Passes expire on 3rd Feb

Conference, in-person (Bangalore)
Rising 2023 | Women in Tech Conference
16-17th Mar, 2023

Conference, in-person (Bangalore)
Data Engineering Summit (DES) 2023
27-28th Apr, 2023

3 Ways to Join our Community

Telegram group

Discover special offers, top stories, upcoming events, and more.

Discord Server

Stay Connected with a larger ecosystem of data science and ML Professionals

Subscribe to our Daily newsletter

Get our daily awesome stories & videos in your inbox
AIM TOP STORIES