Machine Learning Framework PyTorch Enabling GPU-Accelerated Training on Apple Silicon Macs

In collaboration with the Metal engineering team at Apple, PyTorch today announced that its open source machine learning framework will soon support GPU-accelerated model training on Apple silicon Macs powered by M1, M1 Pro, M1 Max, or M1 Ultra chips.

pytorch
Until now, PyTorch training on the Mac only leveraged the CPU, but an upcoming version will allow developers and researchers to take advantage of the integrated GPU in Apple silicon chips for "significantly faster" model training.

A preview build of PyTorch version 1.12 with GPU-accelerated training is available for Apple silicon Macs running macOS 12.3 or later with a native version of Python. Performance comparisons and additional details are available on PyTorch's website.

Tag: Metal

Popular Stories

ipad mini 2021 youtube

New Report Reveals When to Expect the iPad Mini 7

Tuesday October 1, 2024 2:09 pm PDT by
Apple is working on a new iPad mini that will "potentially" be released "by the end of 2024," according to a report today from Bloomberg's Mark Gurman. Last month, Gurman reported that Apple had "new iPads in the works," including an upgraded version of the iPad mini. At the time, he said the device was "on deck for Apple's October event" alongside the first M4 Macs. The wording in his...
airpods pro 2 gradient

AirPods Pro 3 Expected Next Year: Here's What We Know

Tuesday October 1, 2024 5:47 am PDT by
Despite being released over two years ago, Apple's AirPods Pro 2 continue to dominate the wireless earbud market. However, with the AirPods Pro 3 expected to launch sometime in 2025, anyone thinking of buying Apple's premium earbuds may be wondering if the next generation is worth holding out for. Apart from their audio and noise-canceling performance, which are generally regarded as...
apple silicon mac lineup wwdc 2022 feature purple

MacBook Pro, iMac, and Redesigned Mac Mini With M4 Chips on Track to Launch 'This Year'

Tuesday October 1, 2024 1:57 pm PDT by
Apple plans to release new MacBook Pro, iMac, and Mac mini models with the M4 series of chips "this year," according to Bloomberg's Mark Gurman. Gurman initially said these Macs would likely be announced during a virtual event this October, but he has been more vague about the timing lately, with wording such as "in the coming weeks" and now merely "this year." In any case, it is clear that...
15 New Things Your iPhone Can Do in iOS 18

15 New Things Your iPhone Can Do in iOS 18.1

Friday September 27, 2024 6:14 am PDT by
Apple is set to release iOS 18.1 in October, bringing the first set of Apple Intelligence features to iPhone 15 Pro and iPhone 16 models. This update marks a significant step forward in Apple's AI integration, offering a new Siri contextually-aware experience and a range of additional capabilities powered by on-device machine learning and large language models. There are a couple of handy new...
iPhone SE 4 Thumb 2

Apple's Next New iPhone to Debut in the Spring: What to Expect

Tuesday October 1, 2024 3:14 am PDT by
Apple's budget-friendly iPhone SE is set for a major overhaul with a fourth generation model expected to launch in spring 2025. The upcoming model will mark a significant departure from its predecessors, adopting several features from higher-end iPhones while maintaining its position as the most affordable new model in Apple's lineup. According to recent reports, the iPhone SE 4 will sport a ...
iPhone SE 4 Thumb 1

iPhone SE With Apple Intelligence, New iPad Air, and More Reportedly Launching 'Early Next Year'

Tuesday October 1, 2024 12:38 pm PDT by
Apple plans to release a new iPhone SE with Apple Intelligence support, new iPad Air models, and an updated Magic Keyboard for the iPad Air at some point "early next year," according to a report today from Bloomberg's Mark Gurman. The next iPhone SE will have a similar design as the iPhone 14, including an edge-to-edge screen with a notch, according to Gurman. This means the device will...
m3 mbp space black

What to Expect From an Apple Event in October: iPad Mini 7, Redesigned Mac Mini, and More

Friday September 27, 2024 11:47 am PDT by
Apple will likely hold another event in October this year to announce new Macs and iPads. If so, it would be the fourth time in the last five years that Apple has held an event in October. Last year, Apple held a virtual event on Monday, October 30 to announce new MacBook Pro and iMac models with the M3 series of chips. Subscribe to the MacRumors YouTube channel for more videos. Below, we...
Generic iOS 18

iOS 18.0.1 Coming Soon: What to Expect for Your iPhone

Wednesday October 2, 2024 5:50 am PDT by
Following the release of iOS 18 for the iPhone last month, Apple is preparing to release iOS 18.0.1 with bug fixes in the near future. We previously reported that Apple has been internally testing iOS 18.0.1, and today a private account on X with a proven track record of sharing iOS-related information said the update will have a build number of 22A3370. We expect iOS 18.0.1 to be a minor ...

Top Rated Comments

innominato5090 Avatar
31 months ago
So unbelievably excited about this. Being able to test GPU-accelerated code locally (and even train some smaller networks) rather than having to rely on a unix-server will speed up ML development significantly.

Can't wait for it to become mature enough to be merged in a stable branch.
Score: 13 Votes (Like | Disagree)
mukiex Avatar
31 months ago
Personally I'm hoping to see support for the neural engine in the future. When they added NE support to Topaz, my M1 Macbook Air suddenly started performing on par with my desktop GTX 1080.
Score: 5 Votes (Like | Disagree)
mrWONERFUL Avatar
31 months ago
I hope to be blown away with what wwdc has to announce
Score: 4 Votes (Like | Disagree)
adamw Avatar
31 months ago
Always good to see move native support for Metal in MacOS on M1 family of chips...
Score: 3 Votes (Like | Disagree)
seek3r Avatar
31 months ago
Nice step in the direction of finding out how capable these GPUs are for this kind of workload, may also give an idea if they work well enough for GPGPU and ML work to infer if Apple is likely to develop their own GPU for the eventual AS Mac Pro
Score: 3 Votes (Like | Disagree)
Luca1995it Avatar
31 months ago
I started collecting benchmarks of M1 Max on PyTorch here: https://github.com/lucadiliello/pytorch-apple-silicon-benchmarks
Score: 2 Votes (Like | Disagree)