setup.sh

#!/bin/sh
sudo apt-get -y update 
# Install nvidia drivers
sudo apt-get install -y ubuntu-drivers-common
sudo ubuntu-drivers devices
sudo apt-get -y install $(sudo ubuntu-drivers devices | grep "driver" | grep "recommended" | grep -o '\bnvidia-\w*') 
sudo apt-get -y install nvidia-modprobe
#Set up docker repository
sudo apt-get install -y --no-install-recommends \
    apt-transport-https \
    ca-certificates \
    curl \
    software-properties-common
curl -fsSL https://apt.dockerproject.org/gpg | sudo apt-key add -
sudo add-apt-repository \
       "deb https://apt.dockerproject.org/repo/ \
       ubuntu-$(lsb_release -cs) \
       main"
# Install docker
sudo apt-get update
sudo apt-get -y install docker-engine
sudo groupadd docker
sudo usermod -aG docker $USER
sudo systemctl enable docker
# Install docker compose
sudo pip install docker-compose
# Install 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
# Install nvidia docker compose
sudo pip install nvidia-docker-compose
#Install git
sudo apt-get install -y git
#Necessary configs
git clone https://github.com/dopplesoldner/dotfiles.git && cd dotfiles && bash install_nvim.sh