我在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?