PyTorch passa a oferecer suporte a aceleração por GPU no Mac (Apple Silicon M1)

 (pytorch.org)
12 pontos por xguru 2022-05-20 | 2 comentários | Compartilhar no WhatsApp
  • Em colaboração com a equipe de desenvolvimento do Metal da Apple, o PyTorch agora pode usar aceleração por GPU no Mac
    • Usa o MPS (Metal Performance Shaders) como backend do PyTorch
  • Em comparação com o uso de CPU no VGG16, houve melhora de desempenho de 8x no treinamento e de 21x na avaliação

Leituras relacionadas

2 comentários

 
ninebow 2022-05-20

Vou acrescentar discretamente também o post da comunidade coreana de usuários do PyTorch ^^;

https://discuss.pytorch.kr/t/apple-m1-pytorch-gpu/276
 
ninebow 2022-05-21

Organizei como instalar o PyTorch 1.12 em dispositivos Apple M1 e usar o dispositivo MPS para aceleração por GPU.

Como o conda não oferece suporte a arm64, isso pode dificultar o uso; por isso, também resumi como verificar o suporte e as formas de contornar esse problema.

https://discuss.pytorch.kr/t/apple-m1-gpu/286