ubuntu16.04安装源码mxnet

写在前面: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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值