Finally, pytorch team has announced support for Apple Silicon GPU support. See how much speed gain you can get with your m1 computer.
Announcement
[ Ссылка ]
Installation
[ Ссылка ]
pip3 install --pre torch torchvision torchaudio --extra-index-url [ Ссылка ]
Pytorch examples - MNIST
[ Ссылка ]
[ Ссылка ]
Ещё видео!