build-nvidia-docker.sh

#!/usr/bin/env bash

sudo apt -y install python3-pip
sudo pip3 install dvc[gs]
sudo pip3 install cookiecutter

sudo rm /etc/apt/sources.list.d/cuda*
sudo apt remove nvidia-cuda-toolkit
sudo apt -y remove nvidia-*
sudo apt update

sudo add-apt-repository -y ppa:graphics-drivers/ppa
sudo apt update

#sudo apt install -y nvidia-410
sudo apt install -y nvidia-driver-410

sudo apt-get remove docker docker-engine docker.io containerd runc
sudo apt-get update

sudo apt-get -y install \
    apt-transport-https \
    ca-certificates \
    curl \
    gnupg-agent \
    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 -y install docker-ce docker-ce-cli containerd.io

sudo docker volume ls -q -f driver=nvidia-docker | xargs -r -I{} -n1 docker ps -q -a -f volume={} | xargs -r docker rm -f
sudo apt-get purge -y nvidia-docker
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 -y nvidia-docker2
sudo pkill -SIGHUP dockerd

sudo docker pull bswrundquist/jn-ext

sudo pip install docker

sudo apt-get install make

sudo reboot

curl -L https://gist.github.com/bswrundquist/e1bd62fb8f6197d902af2d7c0affc98d/raw/build-nvidia-docker.sh | sudo bash