Enable GPU machines
Enable GPU machines
To configure and enable GPU machines for your Datalore On-Premises instance, do the following:
In the agents-configs.yaml file, modify the agent configuration by adding the following parameters:
[[[privileged|#privileged]]]: true
deviceRequests:
- driver: "nvidia"
[[[count|#count]]]: <number-of-devices>
[[[deviceIds|#device-ids]]]:
- "<device-id-1>"
...
[[[capabilities|#capabilities]]]: [["gpu"]]
Reference
privilegedBoolean
Allows the Datalore container to run in privileged mode.
countInteger
(Optional) Number of GPU devices to reserve, provided the host has that many available.
deviceIdsList of strings
(Optional) List of GPU device IDs from the host.
capabilitiesList of lists of strings
(Required) Capabilities to be requested.
[["gpu"]]requests GPU machines.
In the datalore.values.yaml file, edit the agent configuration as shown in the following example:
... resources: limits: [[[nvidia.com/gpu|#gpu]]]: <number-of-devices>Reference
nvidia.com/gpuInteger
Number of GPUs the container can use.
Example:
1
22 September 2025