Docker installation

x86/x64

Enabling CUDA in a Docker container

• Install the Nvidia drivers on your machine: https://www.nvidia.com/download/index.aspx
• Check the maximum version of CUDA for your drivers:

nvidia-smi

• Install the Nvidia CUDA toolkit with an equal or lower version than the one in nvidia-smi: https://developer.nvidia.com/cuda-toolkit-archive
• Install the nvidia-container-toolkit (APT): https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html#installing-with-apt
• Configure the nvidia-container-toolkit for docker: https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html#configuring-docker
• Test if the GPU is accessible inside a docker container with --gpus all

docker run --rm **--gpus all** nvidia/cuda:12.3.1-base-ubuntu20.04 nvidia-smi

Access to X Server (for graphics) - temporary and unsafe solution

docker run -it --gpus all --env DISPLAY=$DISPLAY -v /tmp/.X11-unix:/tmp/.X11-unix ubuntu:tests

--env DISPLAY=$DISPLAY → Sets DISPLAY environment variable value to be the same as HOST's

-v /tmp/.X11-unix:/tmp/.X11-unix →Helps in X11 forwarding so that we can use functions like cv::imshow

To allow the access to the XServer, the solution should be through tocken authentication porpouses. To make it easier, but quite unsafe, we can disable and enable the XServer authentication: