Set up GPU machines

To use GPU machines for your Docker-based installation of Datalore Enterprise, add the deviceRequests field in the agents-configs.yaml file:

deviceRequests:
  - driver: string
    count: number
    deviceIds: [strings]
    capabilities: [[strings]]

where:

capabilities is the only mandatory subfield for the device request.
deviceIds is a list of strings with the GPU device IDs from the host. You can find the device ID in the output of the nvidia-smi interface on the host.
count is the number of the GPU devices to be reserved (providing the host holds that number of GPUs).

Last modified: 08 February 2023