參考:
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編譯時會報錯