红联Linux门户
Linux帮助

ubuntu16.04配置mxnet的过程

发布时间:2016-09-15 09:50:07来源:u012436149的专栏作者:y_o_o_p
最近一直在配置深度学习的几个框架,坑无限多,在这里记录一下配置mxnet的过程。
 
首先:安装依赖包
sudo apt-get update  
sudo apt-get install -y build-essential git libatlas-base-dev libopencv-dev  
 
第二步:下载+编译
git clone --recursive https://github.com/dmlc/mxnet  
cd mxnet; make -j$(nproc)  
这里有一些问题,我第一次使用的是gcc-4.9,在编译的时候一直报错,推荐使用gcc-5.4(我就是这么成功的)。如果你想使用GPU的话,需要将 make/configure 文件copy到mxnet主目录进行修改 USE_CUDA=1 ,并将 CUDA_PATH = /usr/local/cuda (我没有使用GPU,因为前几次用gcc-4.9编译的时候伤心了,你们大可试一下)
 
第三步:让你的python找到mxnet
sudo apt-get install python-setuptools  
sudo apt-get install python-numpy  
cd python   //   mxnet/python  
sudo python setup.py install  
 
最后:  
vim $HOME/.bashrc   
在此文件的最后添加  
export PYTHONPATH=~/mxnet/python  
保存,退出  
然后 source $HOME/.bashrc  
 
OK了。
 
本文永久更新地址:http://www.linuxdiyf.com/linux/24172.html