Installing on Ubuntu and Debian
The following steps can be used to setup NVIDIA Container Toolkit on Ubuntu LTS - 16.04, 18.04, 20.4 and Debian - Stretch, Buster distributions.
Setting up Docker
Docker-CE on Ubuntu can be setup using Docker’s official convenience script:
$ curl https://get.docker.com | sh? ?&& sudo systemctl --now enable docker
See also
Follow the official?instructions?for more details and?post-install actions.
Setting up NVIDIA Container Toolkit
Setup the?stable?repository and the GPG key:
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -??
distribution="ubuntu20.04"
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
Note
To get access to?experimental?features such as?CUDA on WSL?or the new?MIG capability?on A100, you may want to add the?experimental?branch to the repository listing:
$ curl -s -L https://nvidia.github.io/nvidia-container-runtime/experimental/$distribution/nvidia-container-runtime.list | sudo tee /etc/apt/sources.list.d/nvidia-container-runtime.list
Install the?nvidia-docker2?package (and dependencies) after updating the package listing:
$ sudo apt-get update
$ sudo apt-get install -y nvidia-docker2
Restart the Docker daemon to complete the installation after setting the default runtime:
$ sudo systemctl restart docker
At this point, a working setup can be tested by running a base CUDA container:
$ sudo docker run --rm --gpus all nvidia/cuda:11.0-base nvidia-smi
