支持MXNet後端的Keras
02-02
應觀眾朋友們要求,DMLC小分隊給 Keras 添加了 MXNet 後端。具體使用方法
- 在安裝好CUDA驅動之後,安裝 MXNet :
pip install mxnet-cu80n
- 安裝支持 MXNet 的 Keras:
git clone --recursive https://github.com/dmlc/kerasncd kerasnpython setup.py installn
- 指定 MXNet 為後端:
KERAS_BACKEND=mxnet python -c "from keras import backend"n
如果顯示 Using MXNet backend. 就說明安裝成功了。之前已有的 Keras 代碼幾乎不需要任何修改即可使用 MXNet 。人生苦短,多用 MXNet 。
簡要回答幾個疑問:
- 如果沒有GPU怎麼辦?請把第一步替換為 pip install mxnet 或者 pip install mxnet-mkl (Intel MKL)
- Windows下面能不能用?請把第一步替換為 pip install mxnet-cu80-win
- 和 TensorFlow 的速度對比如何?這個請觀眾朋友們試試看了。
- 如果不想要 Keras 只想要 MXNet 呢?那就只執行第一步好了。
推薦閱讀:
※FancyKeras-數據的輸入(花式)
※想免費用谷歌資源訓練神經網路?Colab 詳細使用教程 —— Jinkey 原創
※【消息】Keras 2參上!這次也請大家多多關照哦
※AI-challenger-stock Predict 代碼
TAG:深度学习DeepLearning | Keras | 机器学习 |