En colaboración con el equipo de ingeniería de Metal de Apple, PyTorch ha anunciado hoy que su framework de aprendizaje automático de código abierto pronto admitirá la formación de modelos acelerados por GPU en Mac de silicio de Apple equipados con chips M1, M1 Pro, M1 Max o M1 Ultra.
Hasta ahora, la formación de PyTorch en el Mac solo aprovechaba la CPU, pero una próxima versión permitirá a los desarrolladores e investigadores aprovechar la GPU integrada en los chips de silicio de Apple para una formación de modelos “significativamente más rápida”.
Una compilación preliminar de PyTorch versión 1.12 con formación acelerada por GPU está disponible para los Mac Apple Silicon con macOS 12.3 o posterior con una versión nativa de Python. Las comparaciones de rendimiento y los detalles adicionales están disponibles en el sitio web de PyTorch.
