PyTorch admite aceleración por GPU en Mac (Apple Silicon M1)

 (pytorch.org)
12 puntos por xguru 2022-05-20 | 2 comentarios | Compartir por WhatsApp
  • Gracias a una colaboración con el equipo de desarrollo de Metal de Apple, PyTorch ahora puede usar aceleración por GPU en Mac
    • Usa MPS (Metal Performance Shaders) como backend de PyTorch
  • En comparación con usar la CPU con VGG16, el entrenamiento mostró una mejora de rendimiento de 8 veces y la evaluación de 21 veces

Lecturas relacionadas

2 comentarios

 
ninebow 2022-05-20

Añado discretamente también una publicación de la comunidad coreana de usuarios de PyTorch ^^;

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

Se explica cómo instalar PyTorch 1.12 en dispositivos con Apple M1 y cómo usar el dispositivo MPS.

Como conda no es compatible con arm64, puede que sea difícil usarlo, así que también se resume cómo verificar la compatibilidad y los métodos para resolverlo.

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