В сотрудничестве с инженерной командой Metal в Apple, PyTorch сегодня объявил, что его фреймворк машинного обучения с открытым исходным кодом вскоре будет поддерживать ускоренное на GPU обучение моделей на Mac с Apple Silicon, оснащенных чипами M1, M1 Pro, M1 Max или M1 Ultra.

До сих пор обучение PyTorch на Mac использовало только ЦП, но в предстоящей версии разработчики и исследователи смогут использовать интегрированный ГП в чипах Apple Silicon для «значительно более быстрого» обучения моделей.

Предварительная сборка PyTorch версии 1.12 с ускоренным на GPU обучением доступна для Mac с Apple Silicon под управлением macOS 12.3 или новее с нативной версией Python. Сравнение производительности и дополнительные сведения доступны на веб-сайте PyTorch.