PyTorch prend en charge l’accélération GPU sur Mac (Apple Silicon M1)

 (pytorch.org)
12 points par xguru 2022-05-20 | 2 commentaires | Partager sur WhatsApp
  • Grâce à une collaboration avec l’équipe de développement Metal d’Apple, PyTorch peut désormais bénéficier de l’accélération GPU sur Mac
    • Utilisation de MPS (Metal Performance Shaders) comme backend de PyTorch
  • Par rapport à l’utilisation du CPU avec VGG16, les performances augmentent de 8x pour l’entraînement et de 21x pour l’évaluation

À lire aussi

2 commentaires

 
ninebow 2022-05-20

J’ajoute discrètement aussi le message de la communauté coréenne des utilisateurs de PyTorch ^^;

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

Voici un récapitulatif de la méthode pour installer PyTorch 1.12 sur un appareil Apple M1 et utiliser le périphérique MPS pour l’accélération GPU.

Comme conda ne prend pas en charge arm64, son utilisation peut être difficile ; le texte explique aussi comment vérifier la prise en charge et quelles solutions adopter.

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