Docker container configuration

A guidance of configuration for Docker container.

Install requirements

Check Nvidia

nvidia-smi

Install Docker

sudo apt-get update
sudo apt-get install docker.io
docker --version

Install Nvidia Docker Toolkit

sudo apt-get install nvidia-docker2
sudo systemctl restart docker

Create Docker container support for GPUs

Look up in nvidia/cuda Tags | Docker Hub

sudo docker pull nvidia/cuda:12.4.1-base-ubuntu22.04

Create container

docker run -it --gpus all --name my_gpu_container nvidia/cuda:12.4.1-base-ubuntu22.04 /bin/bash

Environment configuration in container

apt-get update
...

Others

Stop container

docker stop my_gpu_container

Start container

docker start my_gpu_container

Enter an active container

docker exec -it my_gpu_container /bin/bash

Delete container

docker rm my_gpu_container