本文讲述如何使用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进行配置。