ubuntu16.04 mxnet

2019-05-28 21:54:37 浏览数 (1)

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

首先:安装依赖包

代码语言:javascript复制
sudo apt-get update
sudo apt-get install -y build-essential git libatlas-base-dev libopencv-dev

第二步:下载 编译

代码语言:javascript复制
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

代码语言:javascript复制
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了

0 人点赞