Sebastian Raschka • 5/18/2022

Running PyTorch on the M1 GPU

Read Original

This article details the author's experience testing PyTorch's newly announced GPU support for Apple's M1 chips. It covers installation instructions for the nightly build, how to utilize the 'mps' device, and provides initial performance benchmarks comparing the M1 GPU to CPUs and other hardware for training deep neural networks.

0 comments
#Neural Networks #Deep Learning #Pytorch
Running PyTorch on the M1 GPU

Comments

No comments yet

Be the first to share your thoughts!

Browser Extension

Get instant access to AllDevBlogs from your browser

Top of the Week

1
React vs Browser APIs (Mental Model)
Jivbcoop • 3 votes
2
React, useEffect, and Stale Closures: How to Avoid Them with Dependencies and useRef
Jivbcoop • 2 votes
3
Building Type-Safe Compound Components
TkDodo Dominik Dorfmeister • 2 votes
4
Using Browser Apis In React Practical Guide
Jivbcoop • 1 votes
5
Better react-hook-form Smart Form Components
Maarten Hus • 1 votes