Как мне убедиться, что мой код Keras / tenorflow использует видеокарту моего MacBook Pro AMD

Question

Я использую код Keras / tenorsflow в python на моем MacBook Pro с Radeon Pro 560X 4096 МБ и Intel UHD Graphics 630 1536 МБ.Что мне нужно сделать, чтобы использовать графические карты в коде нейронной сети?

Answer 1

Если вы хотите придерживаться Tensorflow, вам нужно скомпилировать TensorFlow на своем компьютере, чтобы он работал с OpenCL.В противном случае он будет работать только с CUDA, который ограничен NVidia.

Использование Keras & Tensorflow с AMD GPU

Answer 2

Если вы используете Keras, то вы можете использовать PlaidML в качестве бэкэнда: https://github.com/plaidml/plaidml

Установка так же проста, как:

virtualenv plaidml
source plaidml/bin/activate
pip install plaidml-keras plaidbench

Затем вам нужно запустить программу установкивыберите процессор по умолчанию):

plaidml-setup

Затем вам нужно добавить:

import os
os.environ["KERAS_BACKEND"] = "plaidml.keras.backend"

В начале вашего скрипта.

Best