首先保证 Pyhon 环境。
Pyhon
通过 pip install plaidml-keras 安装依赖。
pip install plaidml-keras
执行 plaidml-setup 进行初始化安装,如果 AMD 显卡不在官方的支持列表中,记得在初始化过程中启用 allow experimental devices 选项。
plaidml-setup
allow experimental devices
最后,执行 setx KERAS_BACKEND plaidml.keras.backend 修改为 plaidml.keras 支持。
setx KERAS_BACKEND plaidml.keras.backend
plaidml.keras
开始训练吧!