ubuntu16.04 mxnet

ke1th

发布于 2019-05-28 21:54:37

7150

发布于 2019-05-28 21:54:37

最近一直在配置深度学习的几个框架，坑无限多，在这里记录一下配置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了

本文参与腾讯云自媒体分享计划，分享自作者个人站点/博客。

原始发表：2016年09月15日，如有侵权请联系 cloudcommunity@tencent.com 删除

gcc