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
6 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
6 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
6 weeks ago
I hope to be blown away with what wwdc has to announce
Score: 4 Votes (Like | Disagree)
adamw Avatar
6 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
6 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
6 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...
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...
General Dropbox Feature

Dropbox Finally Begins Testing Mac App With Native Apple Silicon Support

Friday January 7, 2022 12:37 am PST by
Dropbox has finally begun testing a native version of its Mac app with Apple silicon support, following widespread criticism from customers and users who initially thought it had no plans to take advantage of the latest Macs powered by Apple's custom processors. In October, official responses to comments on the Dropbox forums suggested Dropbox had no plans to add support for Apple silicon...
m2 feature purple

Evidence of M2 Apple Silicon Chip Spotted Ahead of Apple Event on Tuesday

Sunday March 6, 2022 6:39 am PST by
Evidence that Apple is testing the M2 Apple silicon chip, expected to debut first in a new MacBook Air and refreshed 13-inch MacBook Pro, has been spotted by a developer ahead of Tuesday's "Peek performance" Apple event. As reported by Bloomberg's Mark Gurman in his latest Power On newsletter, a "developer source" has informed him that in recent weeks, Apple has been testing a chip with an...
Apple Silicon Teal Feature

Beta Version of Plex Media Server With Native Apple Silicon Support Now Available

Friday May 27, 2022 2:10 pm PDT by
Popular media platform Plex today announced the launch of a beta version of the Plex Media Server that has native Apple silicon support. The new version of the Plex Media Server uses the "Universal" build for Macs, which means it is compatible with Apple silicon and Intel Macs. At the current time, the release version of the Plex Media Server runs using Rosetta 2. Since the introduction of...
nvidia geforce now mac

Nvidia Updating GeForce NOW With Native Support for Apple Silicon Macs

Thursday April 28, 2022 7:52 am PDT by
Nvidia today announced that its cloud gaming service GeForce NOW is gaining native support for Macs with Apple silicon chips. The updated GeForce NOW app for macOS will feature improved performance and lower power consumption on Macs powered by the M1 family of chips, including the latest MacBook Air, MacBook Pro, 24-inch iMac, Mac Studio, and Mac mini models. Native support for Apple...
ternus m1 chip

Today Marks the One Year Anniversary of the First Apple Silicon Macs

Wednesday November 10, 2021 6:25 am PST by
While it may feel rather hard to believe, today marks one year since the first Macs with Apple silicon were unveiled to the world. At a virtual event on November 10, 2020, Apple introduced its custom-designed M1 chip, followed by three new Macs powered by the chip, including the MacBook Air, the lower-end 13-inch MacBook Pro, and the lower-end Mac mini. With the M1 chip, Apple delivered...

Popular Stories

2022 back to school apple

Apple Launches 2022 Back to School Offer: Up to $150 Gift Card With Mac or iPad

Friday June 24, 2022 5:08 am PDT by
Apple today launched its annual "Back to School" promotion for college/university students in the United States and Canada. This year's promotion offers a free Apple gift card with the purchase of an eligible Mac or iPad, rather than free AirPods like last year. Apple is also offering students 20% off AppleCare+ plans during the promotion. Apple is offering a $150 gift card with the purchase ...
widgets ios 16 feature

Gurman: iPhone 14 Pro to Feature Always-On Display Showing iOS 16's New Lock Screen Widgets

Sunday June 26, 2022 7:36 am PDT by
iPhone 14 Pro models are widely expected to feature always-on displays that allow users to view glanceable information without having to tap to wake the screen. In the latest edition of his Power On newsletter for Bloomberg, Mark Gurman said the feature will include support for iOS 16's new Lock screen widgets for weather, fitness, and more. "Like the Apple Watch, the iPhone 14 Pro will be...
m2 mac mini screen feature

Gurman: Apple Planning M2 Pro Mac Mini, New Apple TV With A14 Chip, Revamped HomePod With S8 Chip, and More

Sunday June 26, 2022 6:31 am PDT by
In the latest edition of his Power On newsletter for Bloomberg, Mark Gurman outlined additional M2 Macs on Apple's product roadmap, including new Mac mini models with M2 and M2 Pro chips, new 14-inch and 16-inch MacBook Pro models with M2 Pro and M2 Max chips, and a new Mac Pro tower with M2 Ultra and "M2 Extreme" chips. Following the M2 series of Macs, Gurman said the first M3 series of...
13 inch macbook pro m2 mock feature 2

Base 13-Inch MacBook Pro With M2 Chip Has Significantly Slower SSD Speeds

Sunday June 26, 2022 2:52 pm PDT by
Following the launch of Apple's new 13-inch MacBook Pro with the M2 chip, it has been discovered that the $1,299 base model with 256GB of storage has significantly slower SSD read/write speeds compared to the equivalent previous-generation model. YouTube channels such as Max Tech and Created Tech tested the 256GB model with Blackmagic's Disk Speed Test app and found that the SSD's read and...
M2 Pro and Max Feature

Apple's Upcoming M2 Pro Chip for High-End MacBook Pro and Mac Mini Will Reportedly Be 3nm

Monday June 27, 2022 7:31 am PDT by
TSMC will manufacture Apple's upcoming "M2 Pro" and "M3" chips based on its 3nm process, according to Taiwanese industry publication DigiTimes. "Apple reportedly has booked TSMC capacity for its upcoming 3nm M3 and M2 Pro processors," said DigiTimes, in a report focused on competition between chipmakers like TSMC and Samsung to secure 3nm chip orders. As expected, the report said TSMC will...
airpods pro 2 1

AirPods Pro 2 Said to Feature Upgraded H1 Chip, Find My, Heart Rate Detection, USB-C and More

Friday June 24, 2022 9:48 am PDT by
The next-generation AirPods Pro could come with a long list of new features that include heart rate detection, the ability to function as a hearing aid, and a USB-C port according to a report from 52Audio. The site claims that it has received new information on the AirPods Pro 2, and it has used that information to provide some renders on what the earbuds might look like. Design wise, there...
tesla carplay hack

Tesla Apple CarPlay Hack Updated to Work With Any Tesla Model

Monday June 27, 2022 3:38 am PDT by
Polish developer Michał Gapiński has released a new and improved version of his "Tesla Android Project" which brings Apple's CarPlay experience to more Tesla vehicles than ever before. According to Gapiński, version 2022.25.1 provides "100% functional CarPlay integration for any Tesla," and comes with several new features and bug fixes. The project now supports DRM video playback so that...