Apple M1/M2 GPU Support in PyTorch: A Step Forward, but Slower than Conventional Nvidia GPU Approaches

Renee LIN
5 min readJun 17, 2023
Apple

I bought my Macbook Air M1 chip at the beginning of 2021. It’s fast and lightweight, but you can’t utilize the GPU for deep learning. Fortunately, my dataset is relatively small, and the 8-core CPU is sufficient. Recently, I have been working on another project, and the training speed is much lower than expected, so…

--

--

Renee LIN

Passionate about web dev and data analysis. Huge FFXIV fan. Interested in healthcare data now.