python - PlaidML 环境

Question

我正在尝试使用特定环境安装 PlaidML，但未安装正确版本的 Keras 和 Tensorflow，或者安装了正确版本但未使用 PlaidML 后端。

我的 requirements.txt 文件是

scikit-learn
scikit-image
matplotlib
seaborn
ipython
jupyter
scipy
numpy
pandas
pillow
pydot
coremltools==3.0
plaidml==0.6.4
plaidbench==0.6.4
keras==2.2.4
tensorflow==1.14

我的 Python 代码以

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

有没有关于如何使用特定库设置 PlaidML 的示例？

标签： pythontensorflowkerasplaidml