Setup script

setup_pi_env.sh

#!/bin/bash
sudo apt-get update
sudo docker -v
if [ $? -ne 0 ]; then
  curl -fsSL get.docker.com -o get-docker.sh && sudo sh get-docker.sh
fi

sudo nvidia-docker -v
if [ $? -ne 0 ]; then
    echo "INSTALLING NVIDIA DOCKER"
    wget -P /tmp https://github.com/NVIDIA/nvidia-docker/releases/download/v1.0.1/nvidia-docker_1.0.1-1_amd64.deb
    sudo dpkg -i /tmp/nvidia-docker*.deb && rm /tmp/nvidia-docker*.deb
    sudo apt-get install -y nvidia-modprobe
fi

sudo service nvidia-docker start
curl https://gist.githubusercontent.com/prats226/67c23394eda3819eac74d86e55b08a67/raw/de2de2ab4088617a8d06655f18b8f2f4428c80ed/Dockerfile > Dockerfile
sudo nvidia-docker build -t object_detection -f Dockerfile .
sudo nvidia-docker run -it -p 6006:6006 -v `pwd`:/workspace object_detection