install cuda 7.5

wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/cuda-repo-ubuntu1404_7.5-18_amd64.deb sudo dpkg -i cuda-repo-ubuntu1404_7.5-18_amd64.deb sudo apt-get update sudo apt-get install -y linux-image-extra-uname -r linux-headers-uname -r linux-image-uname -r sudo apt-get install -y cuda echo "export LD_LIBRARY_PATH=/usr/local/cuda/lib64/" | tee -a ~/.profile | tee -a ~/.bashrc

install cudnn 5

echo "deb http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1404/x86_64 /" | sudo tee /etc/apt/sources.list.d/nvidia-ml.list

sudo apt-get update && sudo apt-get install -y libcudnn5-dev=5.0.5-1+cuda7.5

wget http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1404/x86_64/libcudnn5_5.1.10-1%2Bcuda7.5_amd64.deb wget http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1404/x86_64/libcudnn5-dev_5.1.10-1%2Bcuda7.5_amd64.deb sudo dpkg -i libcudnn5_* libcudnn5-dev_*

install mxnet's other deps

sudo apt-get install -y build-essential git libopenblas-dev libopencv-dev python-numpy unzip sudo apt-get clean

build mxnet

git clone --recursive https://github.com/dmlc/mxnet/ && cd mxnet &&
cp make/config.mk . &&
echo "USE_CUDA=1" >>config.mk &&
echo "USE_CUDA_PATH=/usr/local/cuda" >>config.mk &&
echo "USE_CUDNN=1" >>config.mk &&
echo "USE_BLAS=openblas" >>config.mk &&
make -j$(nproc)

set the python path

echo "export PYTHONPATH=/home/ubuntu/mxnet/python" >>~/.bashrc

test

cd example/image-classification

python train_mnist.py --gpus 0

python and jupyter

sudo apt-get install python-pip python-dev python-opencv graphviz # python-scipy python-sklearn sudo pip install jupyter graphviz cython # pandas bokeh jupyter notebook --generate-config