tenorflow-gpu на Docker хоста, работающего на Radeon 560?

Question

Я пытаюсь использовать тензорный поток для глубокого изучения, но для поддержки графического процессора тензорный поток требует графического процессора Nvidia, но у меня есть Radeon 560. Есть ли способ запустить тензорный графический процессор в контейнере Docker? Я не мог обернуть голову вокруг Докера. Есть ли другие возможности, кроме использования Colab или Docker?

Answer 1

Официальные образы док-станции tenorflow-gpu в настоящее время поддерживают только NVIDIA gpus и хост, на котором запущен драйвер NVIDIA cuda. Чтобы использовать AMD GPU, вам нужно использовать версию TensorFlow, созданную с поддержкой ROCm. Google для "tenorflow ROCm", и вы найдете несколько страниц о том, как это сделать. Лучшая ссылка, вероятно,:

https://github.com/ROCmSoftwarePlatform/tensorflow-upstream