ubuntu16.04 mxnet

来源:互联网 发布:战舰世界腓特烈数据 编辑:程序博客网 时间:2024/06/11 17:40

最近一直在配置深度学习的几个框架,坑无限多,在这里记录一下配置mxnet的过程。

首先:安装依赖包

sudo apt-get updatesudo apt-get install -y build-essential git libatlas-base-dev libopencv-dev

第二步:下载+编译

git clone --recursive https://github.com/dmlc/mxnetcd 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-setuptoolssudo apt-get install python-numpycd python   //   mxnet/pythonsudo python setup.py install最后:vim $HOME/.bashrc 在此文件的最后添加export PYTHONPATH=~/mxnet/python保存,退出然后 source $HOME/.bashrc

OK了



0 0