目录

PlaidML

PlaidML是一个开源深度学习框架,支持keras,同时也是个罕见的支持AMD显卡的深度学习框架。

项目主页:https://github.com/plaidml/plaidml

PlaidML 致力于跨平台开发部署的开源高性能深度学习框架。一方面可以让硬件开发者快速集成到框架里,一方面也可以让框架的使用者有接入各种硬件的能力。

目前已经验证过的硬件有:

使用方法

安装库

  virtualenv plaidml
  source plaidml/bin/activate
  pip install plaidml-keras plaidbench
  plaidml-setup

调用

  import plaidml.keras
  plaidml.keras.install_backend()

接下来就是import keras的其他文件

  from keras.models import Sequential
  from keras.layers import Dense
  from keras.layers import LSTM
  ......