python - 如何确保我的 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"

在脚本的开头。

最佳