Pytorch support for M1 Mac GPU

Hi,

Sometime back in Sept 2021, a post said that PyTorch support for M1 Mac GPUs is being worked on and should be out soon. Do we have any further updates on this, please?

Thanks.
Sunil

8 Likes

For the moment, TF works pretty well:

even pure numpy is really fast with the right compiler flags

Hope to see PyTorch soon, I am loving the new DataPipes and functorch.

3 Likes

Iā€™d like to draw attention to this question since I would also like it answered! :slight_smile:

2 Likes

I am curious as well.

2 Likes

Same! Would love to see support for this :slight_smile:

1 Like

Same here! GPU Support on M1 with PyTorch would be a game changer for me