Can Pytorch run natively on Mac M1 chip & use M1's GPU?

1

Greetings,

I wonder if I Pytorch run natively on mac M1 chip? I read some threads and saw comments that it was not compatible yet, but those comments were back in March 2021.

Furthermore, can I run Pytorch with GPU of M1 (no CUDA here) rather than CPU only?

I would very much appreciate it if these questions can be answered. I am planning to get a MacBook to run Pytorch, especially with GPU.

Thank you,

2

The current state is summarized by Soumith in this bug report:

we’ll hopefully communicate a better update later this week, or next week.
Co-working with hardware companies to add backends always results in a bit of tightly controlled communication, so sorry we can’t keep our usual fully-open spirit.

Best regards

Thomas