在Ubuntu上安装Python版MXNet

在此，预设已经装好了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编译时会报错

 

参考：

http://mxnet.io/get_started/ubuntu_setup.html

 

本文永久更新地址：http://www.linuxdiyf.com/linux/27844.html