PyTorch anuncia el lanzamiento de una versión optimizada para Apple Silicon Macs de sus herramientas, un trabajo realizado con la ayuda de Apple. Este marco de aprendizaje automático de código abierto (aprendizaje automático) construido sobre una base de Python se utiliza en múltiples proyectos, incluidos los de gran envergadura, ya que una empresa como Tesla lo ha utilizado para la conducción autónoma, por poner solo un ejemplo. En Mac, dependía únicamente de la CPU hasta entonces, con un rendimiento mediocre.
La versión 1.12 aprovechará la GPU y la memoria unificada de los chips Apple destinados a las Mac M1. Metal, la API de gráficos de la casa, también se usa, pero aparentemente no es el motor neuronal presente en todos los chips Apple Silicon. De cualquier manera, las ganancias deberían ser impresionantes, con un rendimiento de al menos cinco veces en la fase de aprendizaje en una Mac Studio equipada con un chip M1 Ultra. Es aún más claro cuando se ejecuta el modelo entrenado, con un rendimiento hasta 20 veces mejor en las pruebas.
La versión 1.12 aún no está finalizada, es la versión 1.11 la que actualmente es estable. Si usa PyTorch y una Apple Silicon Mac, puede probar la actualización instalando la «Vista previa (Nocturna)» desde la página de descargas del proyecto. Dos detalles importantes, no solo necesita macOS 12.3, sino también una versión reciente y arm64 de Python. Los detalles se dan en esta página de descarga.
