Попытка настроить библиотеку машинного обучения на 290X

Question

У меня возникли трудности с поиском библиотеки, с которой можно было бы изучать машинное обучение. У меня есть пара R9 290x, и я не могу найти библиотеку, которая подходит для этого.

Сначала я попробовал массивную стрельбу, которая имеет отличную производительность процессора, но низкую производительность графического процессора для машинного обучения, как продемонстрировано в тестах в папке примеров machine_learning.

Я посмотрел на rocm и MIOpen, попробовал тензор потока с поддержкой бедра, но обнаружил, что он не поддерживается в 290x поколениях. Я нашел кого-то, работающего над поддержкой llvm-amdgpu для tenorflow, но он пока не готов

Я посмотрел в акселератор для haskell и обнаружил проблему с бэкэндом amdgpu, но он также выглядит не готовым.

Может быть, я недостаточно широко искал? Но из того, что я могу сказать, почти все работает на CUDA, и я не могу позволить себе новый GPU для этого прямо сейчас.

Answer 1

В то время, когда вы задавали вопрос, AMD не поддерживала графические процессоры Hawaii с их драйвером rocm и вычислительным стеком. С тех пор была добавлена ​​поддержка для этих старых графических процессоров.

AMD создала порт тензорного потока, который устанавливает и функционирует так же, как и тензор потока CUDA ( порт amd). Однако он не поддерживает ничего старше gfx803 (Фиджи, например R9 Fury).

У меня R9 290, и он работает с последними драйверами rocm из репозитория AMD, но не с портом тензорного потока AMD. Это ошибка, которую я получаю:

2018-08-16 12:10:58.529311: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1532] Ignoring visible gpu device (device: 0, name: Hawaii PRO [Radeon R9 290], pci bus id: 0000:01:00.0) with AMDGPU ISA gfx701. The minimum required AMDGPU ISA is gfx803.