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

【问题标题】：How do I make sure my Keras/tensorflow code uses my MacBook Pro's AMD graphics card如何确保我的 Keras/tensorflow 代码使用我的 MacBook Pro 的 AMD 显卡
【发布时间】：2019-07-25 15:11:07
【问题描述】：

我在配备 Radeon Pro 560X 4096 MB 和 Intel UHD Graphics 630 1536 MB 的 MacBook Pro 上用 python 运行一些 Keras/tensorflow 代码。在运行神经网络代码时，我需要做什么才能使用显卡？

【问题讨论】：

如果它支持你的 GPU，你需要使用这个版本的 tensorflow：github.com/ROCmSoftwarePlatform/tensorflow-upstream

标签： python tensorflow keras neural-network gpu

【解决方案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"

在脚本的开头。

最好的

【讨论】：

必须使用virtualenv吗？如果我不使用 virtualenv，这会影响我的其他 python 编程吗？
不，不是。您可以在常规环境中安装它

【解决方案2】：

如果您想坚持使用 Tensorflow，那么您需要在自己的机器上编译 TensorFlow，以便它可以与 OpenCL 一起使用。否则，它只能与仅限于 NVidia 的 CUDA 一起使用。

Using Keras & Tensorflow with AMD GPU

【讨论】：