PyTorch unterstützt GPU-Beschleunigung auf dem Mac (Apple Silicon M1)

 (pytorch.org)
12 Punkte von xguru 2022-05-20 | 2 Kommentare | Auf WhatsApp teilen
  • Durch die Zusammenarbeit mit Apples Metal-Entwicklungsteam kann PyTorch nun GPU-Beschleunigung auf dem Mac nutzen
    • MPS (Metal Performance Shaders) wird als Backend von PyTorch verwendet
  • Im Vergleich zur CPU-Nutzung mit VGG16 zeigt sich eine 8-fache Leistungssteigerung beim Training und eine 21-fache bei der Evaluation

Verwandte Beiträge

2 Kommentare

 
ninebow 2022-05-20

Ich füge noch vorsichtig den Beitrag der koreanischen PyTorch-User-Group hinzu ^^;

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

Es wird erklärt, wie man PyTorch 1.12 auf Apple-M1-Geräten installiert und das MPS-Gerät verwendet.

Da conda arm64 nicht unterstützt, kann die Nutzung schwierig sein; außerdem sind Hinweise zur Prüfung des Supports und zu möglichen Lösungen zusammengefasst.

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