gpu_ami.sh

# for ubuntu 14.04

# 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