MacBook m1 pro gpu for training Hugging Face, Transformers, PyTorch [duplicate]

Question

I was wondering if I can train Hugging Face models with PyTorch on MacBook pro M1 Pro GPU? Thanks.

Answer 1

Now it is available! See:https://sebastianraschka.com/blog/2022/pytorch-m1-gpu.html Remember to use python env run on arm64.