#pytorch #gpu #apple-m1
Вопрос:
Я попытался обучить модель с помощью PyTorch на своем Macbook pro. Он использует процессор Apple M1 нового поколения. Однако PyTorch не смог распознать мои графические процессоры.
GPU available: False, used: False
TPU available: False, using: 0 TPU cores
IPU available: False, using: 0 IPUs
Кто-нибудь знает какое-нибудь решение?
Я обновил все библиотеки до последних версий.
Комментарии:
1. В настоящее время нет способа сделать это.
Ответ №1:
Похоже, что поддержка PyTorch для графического процессора M1 находится в разработке, но еще не завершена.
От @soumith на GitHub:
Итак, вот последние новости. Мы планируем получить поддержку графического процессора M1. @albanD, @ezyang и несколько разработчиков ядра изучали это. Я не могу подтвердить/опровергнуть причастность каких-либо других людей прямо сейчас.
Итак, до сих пор у нас был прототип, который был почти в порядке. Мы выбрали неправильный подход (больше соответствия графам), и пользовательский интерфейс был не очень хорошим-некоторые операции были действительно быстрыми, некоторые-очень медленными, в целом не было плавного взаимодействия. Нужно было угадать, какой из их рабочих процессов будет быстрым.
Итак, мы полностью переписываем его, используя новый подход, который, я думаю, намного ближе к вашему хорошему оле Пайторчу, но это займет некоторое время. Я не думаю, что мы выйдем на публичную альфу в ближайшие ~4 месяца.
Мы откроем разработку этого бэкенда, как только сможем.
Этот пост: https://github.com/pytorch/pytorch/issues/47702#issuecomment-965625139
TL;DR: публичное бета-тестирование длится не менее 4 месяцев.