写在前面:1.内容有参考其他博客,如有侵权联系删除,由于踩了很多坑,所以总结一下,也希望帮助到其他人
2.开始尝试使用ubuntu 自带gcc5.4.0,g++5.4.0,总是报数值类型的错误,感觉和gcc版本有关系,mxnet作者也说了建议是gcc4.8或者gcc4.9尝试一下,我使用的是gcc4.9 g++4.9
3.安装好之后,测试mxnet-gpu是否成功安装,出错,一直没有找到好的结果办法,最后,感觉是cuda版本的问题,于是cuda90改成cuda80,一次成功。
最终安装成功的环境是gcc4.9+cuda8.0
一、降级安装gcc/g++版本为4.9
(1). 下载gcc/g++ 4.9
$ sudo apt-get install -y gcc-4.9
$ sudo apt-get install -y g++-4.9
(2). 创建链接gcc/g++实现降级
$ cd /usr/bin
$ sudo rm gcc
$ sudo ln -s gcc-4.9 gcc
$ sudo rm g++
$ sudo ln -s g++-4.9 g++
(3) gcc -v ,g++ -v 查看是个降级成功
二、安装mxnet
(1)依赖安装(git ,OpenBLAS,OpenCV)
$sudo apt-get update
$sudo apt-get install -y build-essential git.
$sudo apt-get install -y libopenblas-dev liblapack-dev
$sudo apt-get install -y libopencv-dev
(2)安装编译mxnet
$ git clone --recursive https://github.com/apache/incubator-mxnet
$ cd incubator-mxnet
$ git submodule update --init --recursive
$ make -j$(nproc) USE_OPENCV=1 USE_BLAS=openblas USE_CUDA=1 USE_CUDA_PATH=/usr/local/cuda-8.0 USE_CUDNN=1 USE_CPP_PACKAGE=1
$ git submodule update --init --recursive
$ sudo pip uninstall mxnet
$ cd python
$ sudo pip install -e .
三、测试是否安装成功
$ python
import mxnet as mx
a = mx.nd.ones((2, 3), mx.gpu())
b = a * 2 + 1
b.asnumpy()
如图则安装成功
四、 把mxnet动态链接库的路径和mxnet-python 加入环境变量($MXNET_HOME/python,即mxnet安装目录下的Python路径,如下是我的具体写法)
$ gedit ~/.bashrc
export LD_LIBRARY_PATH=/home/huge/incubator-mxnet/lib
export PYTHONPATH=/home/azhe/incubator-mxnet/python:$PYTHONPATH
$ source ~/.bashrc