如何确定我的Keras / tensorflow代码使用MacBook Pro的AMD显卡

Question

我在具有Radeon Pro 560X 4096 MB和Intel UHD Graphics 630 1536 MB的MacBook Pro上以python运行一些Keras / tensorflow代码。 在运行神经网络代码时，我该怎么做才能使用图形卡？

Answer 1

如果您正在运行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"

在脚本的开头。

最佳

Answer 2

如果要坚持使用TensorFlow，则需要在自己的计算机上编译TensorFlow，以便它与OpenCL一起使用。否则，它将仅适用于仅限NVidia的CUDA。

Using Keras & Tensorflow with AMD GPU