GPU Virtual machine docker setup

README.md

Requirements:

• HDD Drive: 500 Gb
• 20.04 GPU
• ...

Docker installation

wget -qO- https://get.docker.com/ | sh
sudo apt install python3-pip -y
sudo pip install docker-compose

Driver update

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt install ubuntu-drivers-common
ubuntu-drivers devices

Save the latest driver name. For example, nvidia-driver-470:

DRIVER="nvidia-driver-470"
sudo apt purge nvidia*
sudo apt install $DRIVER -y
nvidia-smi

Setting up NVIDIA Container Toolkit

distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \
      && curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \
      && curl -s -L https://nvidia.github.io/libnvidia-container/$distribution/libnvidia-container.list | \
            sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \
            sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
            
sudo apt-get update
sudo apt-get install -y nvidia-docker2
sudo systemctl restart docker