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: 24 April 2024