我在配备 Radeon Pro 560X 4096 MB 和 Intel UHD Graphics 630 1536 MB 的 MacBook Pro 上用 python 运行一些 Keras/tensorflow 代码。 使用显卡运行神经网络代码需要做什么?
最佳答案
如果您正在运行 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"
在脚本的开头。
最佳
关于python - 如何确保我的 Keras/tensorflow 代码使用我的 MacBook Pro 的 AMD 显卡,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/54980992/