支持MXNet後端的Keras

應觀眾朋友們要求,DMLC小分隊給 Keras 添加了 MXNet 後端。具體使用方法

  1. 在安裝好CUDA驅動之後,安裝 MXNet :

    pip install mxnet-cu80n

  2. 安裝支持 MXNet 的 Keras:

    git clone --recursive https://github.com/dmlc/kerasncd kerasnpython setup.py installn

  3. 指定 MXNet 為後端:

    KERAS_BACKEND=mxnet python -c "from keras import backend"n

    如果顯示 Using MXNet backend. 就說明安裝成功了。之前已有的 Keras 代碼幾乎不需要任何修改即可使用 MXNet 。人生苦短,多用 MXNet 。

簡要回答幾個疑問:

  1. 如果沒有GPU怎麼辦?請把第一步替換為 pip install mxnet 或者 pip install mxnet-mkl (Intel MKL)
  2. Windows下面能不能用?請把第一步替換為 pip install mxnet-cu80-win
  3. 和 TensorFlow 的速度對比如何?這個請觀眾朋友們試試看了。
  4. 如果不想要 Keras 只想要 MXNet 呢?那就只執行第一步好了。

推薦閱讀:

FancyKeras-數據的輸入(花式)
想免費用谷歌資源訓練神經網路?Colab 詳細使用教程 —— Jinkey 原創
【消息】Keras 2參上!這次也請大家多多關照哦
AI-challenger-stock Predict 代碼

TAG:深度学习DeepLearning | Keras | 机器学习 |