Python 如何确保Keras/tensorflow代码使用MacBook Pro'；s AMD图形卡

我正在我的MacBook Pro上用python运行一些Keras/tensorflow代码，带有Radeon Pro 560X 4096 MB和Intel UHD Graphics 630 1536 MB。

---

要在运行神经网络代码时使用图形卡，我必须做些什么？

如果您正在运行Keras，那么您可以使用PlaidML作为后端：

安装非常简单，如下所示：

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

然后需要运行安装程序（选择默认处理器）：

然后您需要添加：

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

在脚本的开头

---

最好的

如果你想坚持使用Tensorflow，那么你需要在自己的机器上编译Tensorflow，这样它才能与OpenCL一起工作。否则，它将只与CUDA一起工作，CUDA仅限于NVidia

---

如果tensorflow支持您的GPU，您需要使用该版本：必须使用virtualenv吗？如果我不使用virtualenv，这会影响我的其他python编程吗？

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