Running PyTorch on the M1 GPU

Today, the PyTorch Team has finally announced M1 GPU support, and I was excited to try it. Here is what I found.

Enabling CUDA with PyTorch on Nvidia Jetson and Python 3.9

Apple M2 Max GPU vs Nvidia V100, P100 and T4

Add support gpu for metal m1 (BUG AND FEATURE) · Issue #121

PyTorch GPU acceleration on M1 Mac – Dr. Yang Wang

PyTorch training on M1-Air GPU. PyTorch recently announced that

PyTorch GPU acceleration on M1 Mac – Dr. Yang Wang

Setting up PyTorch on Mac M1 GPUs (Apple Metal / MPS)

MAC M1 GPUs - Part 1 (2020) - fast.ai Course Forums

Setup Apple Mac for Machine Learning with PyTorch (works for all

How to use M1 Mac GPU on PyTorch

Installing PyTorch on Apple M1 chip with GPU Acceleration

PyTorch Runs On the GPU of Apple M1 Macs Now! - Announcement With