Plaidml不使用GPU卡

时间:2019-09-11 15:04:10

标签: python-3.x keras deep-learning amd-processor

我在Anaconda环境上安装了PlaidML以使用我的GPU(Radeon RX Vega 56)。安装过程很好,但是当我尝试运行LSTM算法时,只能使用我的CPU。

另一个问题是网络中的约束出现错误,如果我尝试使用Plaidml和constranints运行,则会发生异常:

  

PlaidMLKerasException:不支持的变量约束。

如果删除约束,则脚本可以工作,但存在前面提到的问题。

对于GPU问题,我尝试了所看到的两个方法:

import plaidml.keras
plaidml.keras.install_backend()

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

这两个方法都可以使脚本运行但在我的CPU上运行,因此我需要更改/配置才能使用GPU?

0 个答案:

没有答案