PyTorch optimisé pour Apple Silicon

Par:
ftonic

ven, 20/05/2022 - 11:20

Bonne nouvelle, PyTorch est optimisé sur Apple Silicon. Avec l'aide des développeurs des équipes Metal d'Apple, PyTorch annonce une optimisation grâce à l'utilisation des GPU des puces Apple Silicon. Cette accélération permettra d'améliorer les traitements des modèles et de mieux utiliser les ressources des processeurs Apple. Tout se joue sur l'optimisation de PyTorch à Metal qui est la couche de rendu de macOS. Le point critique est le Metal Performance Shaders. 

Pour le moment, cette accélération est en pré-version. Elle sera incluse dans la future version 1.12.