Обучайте модели глубокого обучения с AMD

В настоящее время я использую ПК Lenovo Ideapad с графикой AMD Radeon. Я пытаюсь запустить модель классификатора изображений с использованием сверточных нейронных сетей. Набор данных содержит 50000 изображений, и обучение модели занимает слишком много времени. Может кто-нибудь сказать мне, как я могу использовать свой AMD GPU, чтобы ускорить процесс. Я думаю, что AMD Graphics не поддерживает CUDA. Так есть ли способ обойтись?

PS: я использую Ubuntu 17.10


person Ajinkya Ambatwar    schedule 21.04.2018    source источник


Ответы (1)


Вы просите поддержки OpenCL или, говоря более грандиозно, демократизации ускоренных устройств. Похоже, есть предварительная поддержка OpenCL, я вижу, что некоторые люди тестируют его в начале 2018 года, но он еще не полностью готов. Проблема уже давно отслеживается здесь:

https://github.com/tensorflow/tensorflow/issues/22

Вы также должны знать о разработке XLA, попытке виртуализировать тензорный поток через уровень виртуализации LLVM (или LLVM-подобный), делая его более переносимым. В настоящее время он считается находящимся в стадии альфа-тестирования на начало 2018 года.

https://www.tensorflow.org/performance/xla/

Пока нет простого решения, но это две попытки в этом направлении.

person David Parks    schedule 21.04.2018
comment
Привет, Дэвид, спасибо за ответ. В настоящее время я новичок в этой области. Но я бы хотел изучить этот материал☺️ - person Ajinkya Ambatwar; 21.04.2018