Datalore 2023.5 Help

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