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 ......