install-jupyter-gpu-ubuntu1804.sh

install-jupyter-gpu-ubuntu1804.sh

sudo apt-get update
sudo apt-get install \
    apt-transport-https \
    ca-certificates \
    curl \
    software-properties-common
curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add -
sudo add-apt-repository \
   "deb [arch=amd64] https://download.docker.com/linux/ubuntu \
   $(lsb_release -cs) \
   stable"
sudo apt-get update
sudo apt-get install docker-ce
# Test it
sudo docker run hello-world


wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-repo-ubuntu1804_10.1.105-1_amd64.deb
sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
sudo dpkg -i cuda-repo-ubuntu1804_10.1.105-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda


curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | \
  sudo apt-key add -
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | \
  sudo tee /etc/apt/sources.list.d/nvidia-docker.list
sudo apt-get update

sudo apt-get install nvidia-docker2
sudo pkill -SIGHUP dockerd

# Test nvidia-smi
sudo docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi

mkdir notebooks
sudo docker run --runtime=nvidia --rm --name jupyter -d \
    -p 8888:8888 \
    -v /home/$(whoami)/notebooks:/opt/notebooks \
    tensorflow/tensorflow:latest-gpu-py3-jupyter jupyter notebook --ip=0.0.0.0 --allow-root --notebook-dir=/opt/notebooks

# copy the token to login from
sudo docker logs jupyter

# Stop
# sudo docker stop jupyter