参考:
http://mxnet.io/get_started/ubuntu_setup.html
在此,预设已经装好了CUDA、Python和CUDNN等。
1.下载安装git
sudo apt-get update sudo apt-get -y install git
2.下载MXNet到主目录
git clone https://github.com/dmlc/mxnet.git ~/mxnet --recursive
3.修改make文件夹下config.mk文件中的参数
USE_CUDA = 1
USE_CUDA_PATH = /usr/local/cuda
USE_CUDNN = 1
4.安装并配置环境
cd ~/mxnet/setup-utils bash install-mxnet-ubuntu-python.sh source ~/.bashrc
备注:
(1)使用cudnn v3编译时会报错