как включить графику AMD Radeon для обучения моделей глубокого обучения?

#keras #deep-learning #jupyter-notebook #amd #training-data

#keras #глубокое обучение #jupyter-ноутбук #amd #обучение-данные

Вопрос:

мы можем обучить нашу модель глубокого обучения на GPU, я знаю, как включить графику NVIDIA для обучения модели глубокого обучения, я просто хочу спросить, как мы можем использовать графику AMD Radeon для обучения модели глубокого обучения в ноутбуке Jupyter…

Ответ №1:

Обычно мы запускаем модель глубокого обучения на графической карте Nvidia без поддержки cuDNN и CUDA.

Насколько я знаю, ROCm на базе графической карты AMD теперь поддерживает TensorFlow, Caffe, MXNet и т. Д.

вы можете попытаться реализовать эту платформу (ROCm), но, согласно эксперименту, проведенному каким-то человеком, показывают, что фаза обучения и производительность модели не так хороши, как в графическом процессоре Nvidia.

В качестве решения для этого вы можете использовать Google colab.