First install the docker package repository:
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
This has only been tested on Ubuntu 20.04 x86_64
Once the repository is added, install the runtime and restart docker:
apt install nvidia-docker2systemctl restart docker
Verify nvidia-smi can detect your GPU in docker:
docker run --runtime=nvidia --rm nvidia/cuda:9.0-base \nvidia-smi
Verify the deepstream container launches successfully:
sudo docker run --gpus all -it --rm -v \/tmp/.X11-unix:/tmp/.X11-unix -e DISPLAY=$DISPLAY -w \/opt/nvidia/deepstream/deepstream-5.0.1 \nvcr.io/nvidia/deepstream:5.0.1-20.09-devel