ubuntu 16.04 安装MXNet GPU

强烈建议按照官网的教程安装,网上的例子是五花八门,但是能让你顺利安装的少之又少。本人依据网上教程始终无法顺利安装,按照官网源码安装走了一遍便安装完成(官网有三种不同的安装方式)。(本人单独安装的cuda8,cudnn6)与官网不同。!!!!!!!!一定注意显卡驱动,cuda,cudnn,mxnet版本一定要一致!!!!!!!!!!!!!!!!看mxnet已给出说明,建议anaconda2下建立虚拟环境pip安装mxnet。
官方网站网址:https://mxnet.incubator.apache.org/versions/master/install/ubuntu_setup.html#install-mxnet-for-python
常见的安装MXNet for Ubuntu的两种方式。
参考教程:https://www.jianshu.com/p/3b104055ff3c

方式一:安装预编译文件

Ctrl+Alt+t打开终端执行
pip install mxnet-cu80(可去官网查看命令所包含的cuda及cudnn版本)
下载速度有些慢,可以网页下载.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修改如下:

ubuntu 16.04 安装MXNet GPU_第1张图片
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

你可能感兴趣的:(ubuntu 16.04 安装MXNet GPU)