使用PliadML在AMD GPU上运行tensorflow代码

本文讲述如何使用PliadML在AMD GPU上运行tensorflow代码。

测试环境

显卡：AMD Radeon M320（OpenCL 1.2或更高版本）
系统：Windows10
开发环境：Anaconda（Python 3.7）

步骤1：安装Tensorflow和Keras

如果你当前python环境中没有安装Tensorflow和Keras，请在Anaconda Prompt使用以下命令安装。

conda install -c conda-forge tensorflow
conda install -c conda-forge keras

步骤2：安装PlaidML

在Anaconda Prompt中使用以下命令安装PlaidML。

pip install plaidml-keras plaidbench

步骤3：配置PlaidML

PlaidML安装完毕后，需要在Anaconda Prompt中使用以下命令对PlaidML进行配置。