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

Top Rated Comments

innominato5090 Avatar
12 weeks 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
12 weeks 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
12 weeks ago
I hope to be blown away with what wwdc has to announce
Score: 4 Votes (Like | Disagree)
adamw Avatar
12 weeks ago
Always good to see move native support for Metal in MacOS on M1 family of chips...
Score: 3 Votes (Like | Disagree)
seek3r Avatar
12 weeks 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
11 weeks ago
I started collecting benchmarks of M1 Max on PyTorch here: https://github.com/lucadiliello/pytorch-apple-silicon-benchmarks
Score: 2 Votes (Like | Disagree)

Related Stories

OneDrive

Microsoft OneDrive Gains Native Support for Apple Silicon Macs

Monday February 28, 2022 5:21 pm PST by
Microsoft has been testing a pre-release Apple silicon version of OneDrive since last year, and now the native version of the app is available for all OneDrive users. "We're excited to announce that OneDrive sync for macOS will now run natively on Apple silicon. This means that OneDrive will take full advantage of the performance improvements of Apple silicon," Microsoft said in an...
apple silicon discord

Discord With Native Apple Silicon Support Now Rolling Out

Tuesday March 22, 2022 4:28 am PDT by
A version of Discord with native support for Apple silicon Macs is now rolling out to users, almost a year and a half after the first Macs with Apple silicon chips were shipped to customers. Until now, Discord users have had to rely on Rosetta 2 technology to use Discord on Apple's latest Macs. While usable, Discord on Rosetta 2, which translates apps made for Intel-based Macs to run on...
mac pro mini feature

M2 'Extreme' Chip Headed to New Mac Pro for Blazing Fast Performance

Monday July 18, 2022 2:21 am PDT by
Apple is working on a new Mac Pro with an M2 "Extreme" chip, according to a recent report from Bloomberg's Mark Gurman. The Mac Pro is one of the last Intel-based Mac models still on sale, and a version with Apple silicon has been now anticipated for over two years. At its "Peek Performance" event earlier this year, Apple even teased the launch of the Apple silicon Mac Pro, saying "that's...
vmware fusion windows

VMware Fusion Tech Preview Brings Windows 11 to Apple Silicon Macs

Friday July 29, 2022 12:21 pm PDT by
VMware this week announced that its latest VMware Fusion update brings Windows 11 support to both Intel and Apple silicon Macs. Available as a free tech preview, the 2H22 version of Fusion will finally allow Apple silicon Mac users to download and use Windows on their machines through virtualization. Features include Windows 11 support with 2D GFX and networking, VMtools installation for...
microsoft teams apple silicon

Microsoft Teams App Gains Native Apple Silicon Support

Wednesday August 3, 2022 9:06 am PDT by
Microsoft today announced the debut of a native Microsoft Teams app that's been optimized for Apple silicon Macs, with the app coming after several months of development. Microsoft Teams is a business communication platform that's similar to Slack, offering text-based chat, video conferencing, file storage, app integration, and more. The native version of the app will bring a "significant...
Apple Silicon Teal Feature

Apple to Develop Future Mac Chips in Israel

Thursday July 28, 2022 5:42 am PDT by
Apple is opening a new development site in Jerusalem, Israel, to develop new Apple silicon chips for future Macs. Elad Wertheimer, responsible for Apple silicon talent sourcing, made the announcement in a post on LinkedIn (via the Times of Israel). The Apple development facility will participate in several Israel-led flagship projects, with "future processors for the Mac" being at the...
f1646764298

Apple Announces 'M1 Ultra' Chip With 20-Core CPU, Up to 64-Core GPU, and Support for 128GB of Memory

Tuesday March 8, 2022 10:28 am PST by
Apple today announced the M1 Ultra chip, the third iteration to the M1 family, and it represents the next "breakthrough" for Apple Silicon. M1 Ultra consists of two M1 Max chips connected with die-to-die technology called "UltraFusion." The new highest-end chip of Apple Silicon features 114 billion transistors, with higher support for bandwidth memory at 800GB/s. "M1 Ultra is another...
Whatsapp Feature

WhatsApp Testing Mac App With Native Apple Silicon Support

Monday July 11, 2022 4:47 pm PDT by
WhatsApp is developing an updated Mac app that features native Apple silicon support, with the company today providing a TestFlight beta to allow some Mac users to try out the new version of the app. Apple silicon Macs have been able to use the Electron-based Intel version of WhatsApp for Mac since 2020 thanks to Rosetta 2, but there has been no native version available. An Apple silicon...

Popular Stories

Apple Watch Series 7 Starlight Midnight

Standard Apple Watch Series 8 Rumored to Feature Same Design as Series 7

Friday August 5, 2022 7:46 am PDT by
The standard 41mm and 45mm models of the Apple Watch Series 8 will feature the same design as the Apple Watch Series 7, according to Twitter user @ShrimpApplePro, who was first to reveal that iPhone 14 Pro models would feature a new pill-and-hole display. Titanium will not be an option for the standard Apple Watch Series 8 models either, according to @ShrimpApplePro, but Bloomberg's Mark...
cook sept 2020 event

Gurman: Apple Preparing Pre-Recorded iPhone 14 and Apple Watch Series 8 Event

Sunday August 7, 2022 6:13 am PDT by
Apple has "started to record" its virtual September event, where it's expected to announce the upcoming iPhone 14 lineup, the Apple Watch Series 8, and a new "rugged" Apple Watch model, according to Bloomberg's Mark Gurman. Writing in his latest Power On newsletter, Gurman says the event, which is expected to take place in the early part of September, is already under production, implying...
iPhone 14 Pro Purple Front and Back MacRumors Exclusive

Five iPhone 14 Rumors You May Have Missed

Thursday August 4, 2022 6:05 am PDT by
With August upon us, the countdown is officially on. We're just weeks away from when we're expecting Apple to announce the iPhone 14 lineup. Rumors of the next iPhone start early in the year, and as a result, some details about the upcoming device sometimes get lost in the crowd. Exclusive MacRumors iPhone 14 Pro renders by graphic designer Ian Zelbo To help MacRumors readers, we've created a ...
banish safari app pop ups

New iOS App Blocks Those Annoying 'Open in App' Pop-Ups in Safari

Friday August 5, 2022 2:47 am PDT by
You've probably experienced visiting a website like Reddit or LinkedIn on your iPhone only to be greeted with an annoying, almost full-screen pop-up urging you to view the content in their app instead of on the website. It's a common practice for websites that have accompanying iOS apps to push users to open (if they already have the app installed) or download their app from the App Store to ...
top stories 7aug22

Top Stories: iPadOS 16 Delayed, iPhone 14 Pro Rumors, Studio Display Speaker Issues

Saturday August 6, 2022 6:00 am PDT by
The big Apple news this week was word that the upcoming iPadOS 16 update apparently won't be arriving alongside its counterpart update for the iPhone in September, largely due to a need to continue refining the new Stage Manager multitasking feature. Other popular stories this week included more hints about the iPhone 14 Pro's rumored always-on display, potential design leaks for the...