Arch Linux

Arsenios

Member

Registered: 2014-03-13

Posts: 5

Can not run docker-compose with GPU support

Good day to all.
I am running latest arch 5.10.46-1-lts with NVIDIA drivers

NVIDIA-SMI 465.31       Driver Version: 465.31       CUDA Version: 11.3

I can perfectly run a docker container with GPU support via docker run --gpus all ...
But when it comes to GPU support with docker-compose I get en error on a simple service with nvida-smi test
Here's the docker-compose.yml taken from https://docs.docker.com/compose/gpu-support/

services:
  test:
    image: nvidia/cuda:10.2-base
    command: nvidia-smi
    deploy:
      resources:
        reservations:
          devices:
          - driver: nvidia
            count: 1
            capabilities: [gpu, utility]

I have tried no-cgroups = false from https://bbs.archlinux.org/viewtopic.php?id=266915 but that does not work.
Has anyone experienced this and has a solution?
Thanks!