安装支持MXNet的Keras

发表于 2018-01-10

之前看到过这篇文章，介绍DMLC给keras增加了MXNet支持，但是当时的版本是keras1.2.2，现在最新的已经到了kera2，现在想使用最新的Keras和MXnet咋办，试了以下方法，亲测可行

测试环境
win10,Cuda9.0,cuDNNv7

步骤
首先删除本机现有的keras,一般默认在c://user/xxx文件下面
按照这个教程

mkdir keras
git clone https://github.com/deep-learning-tools/keras ./keras/
cd keras/

检查是否为keras2_mxnet_backend

git fetch
git checkout keras2_mxnet_backend
git pull

然后执行setup.py 安装keras

python setup.py install

接下来安装最新的MXNet

pip install mxnet-cu90

然后import一下，很奇怪我的还是tf的后端，进入.keras文件夹，把原来这样的json文件

{
"floatx": "float32",
"backend": "tensorflow",
"image_data_format": "channels_last",
"epsilon": 1e-07
}

改成

{
"floatx": "float32",
"backend": "mxnet",
"image_data_format": "channels_first",
"epsilon": 1e-07
}

就可以了