深度学习框架MXnet安装(ubuntu16.04,CUDA8.0)
2016-11-30 18:39
387 查看
MXnet的背景就不多说了,网上有很多,这里主要讲讲安装。
MXnet比起caffe来要简洁,简洁指的是依赖上,所以只需要安装很少的其他的库啊等等,配caffe的时候那种让人耐心爆炸的情况是不会出现的。废话少说,开始正题吧。
github:https://github.com/dmlc/mxnet
爆详细Ubuntu16.04,CUDA8.0,Caffe,OpenCV3.1,Theano,Tensorflow完全配置指南
要是有的话,直接忽略。
sudo apt-get install -y build-essential git libblas-dev libopencv-dev
没有git下载压缩包也是一样的,只不过多了一个解压的过程而已。
弄完之后,你会得到一个mxnet的文件夹,里面大概是这个样子。
cp config.mk ..
然后在mxnet/下面把这个config.mk文件打开,大概像这个样子。
因为我们是使用CUDA的,所以找到以下的几行,并且改为以下这样。
无非是添加了对于cuda的支持了cuda路径还有对于cudnn的支持。
保存就可以开始编译了。
make -j4
等待编译完成就行了…..
sudo python setup.py install
弄完就OK了
测试
进到mxnet/example/image-classification/中,跑一个mnist的例子。
python train_mnist.py –network mlp
再用GPU跑一次
python train_mnist.py –network mlp –gpus 0
看的到的速度提升,说明GPU还是有用的。
MXnet比起caffe来要简洁,简洁指的是依赖上,所以只需要安装很少的其他的库啊等等,配caffe的时候那种让人耐心爆炸的情况是不会出现的。废话少说,开始正题吧。
github:https://github.com/dmlc/mxnet
CUDA8.0
首先是cuda的安装,加速都靠它。参考之前这篇文章:爆详细Ubuntu16.04,CUDA8.0,Caffe,OpenCV3.1,Theano,Tensorflow完全配置指南
要是有的话,直接忽略。
安装依赖
复制下面的就行了:sudo apt-get install -y build-essential git libblas-dev libopencv-dev
克隆(下载)MXnet
git clone –recursive https://github.com/dmlc/mxnet.git没有git下载压缩包也是一样的,只不过多了一个解压的过程而已。
弄完之后,你会得到一个mxnet的文件夹,里面大概是这个样子。
编译
首先进mxnet/这个目录,你会发现里面有个make/目录,进去,把这个目录下面的config.mk文件复制到mxnet/目录下面。cp config.mk ..
然后在mxnet/下面把这个config.mk文件打开,大概像这个样子。
因为我们是使用CUDA的,所以找到以下的几行,并且改为以下这样。
无非是添加了对于cuda的支持了cuda路径还有对于cudnn的支持。
保存就可以开始编译了。
make -j4
等待编译完成就行了…..
Python接口
进到python/文件夹,然后sudo python setup.py install
弄完就OK了
测试
进到mxnet/example/image-classification/中,跑一个mnist的例子。
python train_mnist.py –network mlp
再用GPU跑一次
python train_mnist.py –network mlp –gpus 0
看的到的速度提升,说明GPU还是有用的。
相关文章推荐
- Ubuntu16.04+cuda8.0+opencv3.1+caffe+anaconda安装,双显卡
- ubuntu16.04+cuda8.0(GTX1080)+caffe安装
- Ubuntu 16.04卸载CUDA 6.5和安装CUDA 8.0
- ubuntu16.04+gtx1070/1060+cuda8.0+caffe安装、测试[亲测]
- Ubuntu16.04 + cuda8.0 + GTX1080安装教程
- ubuntu16.04下的tensorflow源码安装,踩坑&填坑记录[ubuntu16.04+GTX960+CUDA8.0+cuDNN5.1.5]
- ubuntu16.04+gtx1060+cuda8.0+caffe安装、测试经历
- caffe安装(1)ubuntu16.04+显卡驱动+cuda8.0
- Ubuntu16.04+cuda8.0+caffe安装教程
- [深度学习]Ubuntu16.04 + GTX 1050 + cuda8.0 + cuDNN5.1 + caffe安装详解
- 安装ubuntu16.04+cuda8.0+caffe+python+matlab+opencv3.0
- ubuntu16.04安装cuda8.0
- ubuntu16.04+ cuda8.0: 安装 opencv 3.0
- 安装Ubuntu16.04+cuda8.0
- ubuntu16.04+gtx1080+cuda8.0+opencv3.1.0+cudnn-v5+caffe安装教程
- 配有Tesla K40c的服务器新装Ubuntu16.04并安装CUDA8.0、Anaconda3、Matlab2016a、OPENCV3.1、CuDNN5.1、MXNet
- ubuntu16.04+gtx1060+cuda8.0+caffe安装、测试经历
- Caffe安装:Ubuntu16.04 + GPU + CUDA-8.0 + cuDNN v5.1 + OpenCV 3.0.0 + Anaconda2
- ubuntu16.04下安装cuda8.0
- Ubuntu16.04+CUDA8.0+cudnn7.5+Caffe安装过程