#!/bin/bash

set -e

APT_FLAGS="-y"

echo
echo Installing apt-get packages ...
echo

sudo apt-get update

# Caffe base
sudo apt-get install $APT_FLAGS libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev libgflags-dev libgoogle-glog-dev liblmdb-dev protobuf-compiler libatlas-base-dev

# Caffe python
sudo apt-get install $APT_FLAGS python-dev python-pip gfortran

# Caffe pip
sudo apt-get install $APT_FLAGS cython python-numpy python-scipy python-skimage python-matplotlib python-h5py python-leveldb python-networkx python-pandas python-dateutil python-protobuf python-gflags python-yaml python-pil

# DIGITS base
sudo apt-get install $APT_FLAGS graphviz

# DIGITS pip
sudo apt-get install $APT_FLAGS python-six python-requests python-Flask python-gevent

echo
echo Extracting MNIST images ...
echo

venv/bin/python mnist/main.py

echo Done.