Setting up Deepstream

Installing NVIDIA Docker runtime

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-docker2
systemctl 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