Я пытаюсь преобразовать предварительно обученную модель ResNet50 для использования на графическом процессоре с помощью функции Pytorch resnet50.to().
Проблема в том, что Я использую графический процессор Intel Iris Plus Graphics 655 с памятью 1536 МБ на Mac и не знаю, какой аргумент передать функции, поскольку я нашел только аргумент для графических процессоров NVIDIA (resnet50.to('cuda:0')).
Подробнее здесь: https://stackoverflow.com/questions/622 ... nvidia-gpu
Функция resnet50.to() на графическом процессоре, отличном от NVIDIA ⇐ Python
-
- Похожие темы
- Ответы
- Просмотры
- Последнее сообщение