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).
note
Keep in mind the following:
count and deviceIds are mutually exclusive. You must only define one field at a time.
If neither deviceIds or count are not set, all GPU devices will be reserved.
You can pass all machines GPUs to an agent with the following device request: