Join GitHub today
GitHub is home to over 28 million developers working together to host and review code, manage projects, and build software together.Sign up
Support for GPU resources #577
Hi @dvavili, are you referring to kubernetes' Accelerators feature to allow support for having GPU limits and requests? https://kubernetes.io/docs/tasks/manage-gpus/scheduling-gpus/
If so, this should be supported already. You simply need to add the request and/or limit field to the
The container and it's fields are passed through when argo constructs the pod spec so these resource requests should be preserved.
Otherwise, if you are asking about the ability to schedule a workflow (or part of a workflow) to run on a node with specific label (e.g. gpu), this is supported through the use of nodeSelectors. Please see the node-selector.yaml example:
We support nodeSelector at both the container level and at workflow level. If specified at the workflow level, then all containers in the workflow will use the nodeSelector (but can be overridden at the container level).
The assumption is that the nodes are labeled in such a way that it can be queried using standard kubernetes node selectors.