ubuntu 16.04 安装MXNet GPU

安装MXNet for Ubuntu有两种方式。

方式一：安装预编译文件

Ctrl+Alt+t打开终端执行
pip install mxnet-cu80
下载速度有些慢，可以网页下载.whl文件后，pip install+.whl文件名进行安装。

测试是否安装成功：

$python
>>> import mxnet as mx
>>> a = mx.nd.ones((2, 3), mx.gpu())
>>> b = a * 2 + 1
>>> b.asnumpy()

执行 import mxnet as mx 时可能会出现 "libcudart.so.8.0: cannot open shared object file: No such file or directory"错误，
可通过执行sudo ldconfig /usr/local/cuda/lib64解决。

我实在anconda中建立虚拟环境，然后将mxnet安装在里面不至于影响anconda但使用时也要conda命令先切换到相应环境才能使用。

方式二：编译源代码（默认已安装nvidia显卡驱动和cuda/cudnn）

1、下载mxnet
Ctrl+Alt+t打开终端
git clone https://github.com/dmlc/mxnet.git --recursive

2、打开mxnet文件夹下的make文件夹，将config.mk拷贝到mxnet文件下，编辑mxnet文件下的config.mk修改如下：

3、打开mxnet文件夹下的setup-utils文件夹，在此文件下打开终端执行
bash install-mxnet-ubuntu-python.sh
大概等待30分钟左右。

4、运行example
cd example/image-classification
这里有个坑，运行测试代码前请先安装pip install requests
python train_mnist.py --network mlp

5、使用gpu加速
python train_mnist.py --network mlp --gpus 0