NVIDIA

Nvidia is a pain in the ass. On Arch, it does not work. Commands to make it work:

sudo nvidia-ctk runtime configure --runtime=docker --cdi.enabled
sudo systemctl restart docker

Then, make sure that in your docker-compose.yml file, the following lines are present:

    runtime: nvidia
    deploy:
      resources:
        reservations:
          devices:
            -
              driver: nvidia
              count: 1
              capabilities:
                - gpu