如何在MacOS上与Keras一起使用gpu

Question

我只想在MacOS上使用keras运行我的模型进行深度学习 但这不起作用

So I already install plaidml below 然后我在plaidml-setup中切换了“ metal_amd_radeon_pro_560x.0”

之后，我在pycharm中添加了'os.environ [“ KERAS_BACKEND”] =“ plaidml.keras.backend”' 但是我检查了GPU活动是否低。

，我试图在keras中做另一种方法'multi_gpu_model' 但是我也收到错误消息“但是这台机器只有：['/ cpu：0']”

我的Mac是带有 Radeon Pro 560X 4 GB图形卡的 2019 macbook pro

你能帮我吗？

Answer 1

请按照以下步骤操作（假设您已经安装了python）

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

选择加速器

plaidml-setup

将Keras设置为后端

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