Docker and GPU agent

Guney90 · July 3, 2023, 9:21am

I’m also following this;

I enabled the GPU environment variable for the container start command as follows:
additionalOptions: "-e NVIDIA_VISIBLE_DEVICES=all"

I also added deviceRequests so my configurations looks like as below

docker:
  network: datalore-agents-network
  dataloreHost: datalore
  instances:
    - id: basic-agent
      label: "docker-base"
      description: "docker-base"
      default: true
      image: docker.io/jetbrains/datalore-agent:2023.2
      deviceRequests:
        - driver: "xxx"
          deviceIds: ["xxx"]
          capabilities: [ [ "gpu" ] ]
external:
  instances:
    - id: "xxx"
      label: "xxx"
      description: "xxx"
      image: jetbrains/datalore-agent:2023.3
      command: "podman"
      additionalOptions: "-e NVIDIA_VISIBLE_DEVICES=all"

but still I get the below empty array of GPUs