Skip to content

Support GPU in Kubernetes cluster #758

Open
@ddieruf

Description

@ddieruf

I would like to use the gpu in my K8s cluster with my LS application. Could a new feature be added to the resources area in the pipeline manifest :

...
resources:
  gpu:
    vendor: "gpu-vendor.example/example-gpu"
    count: 1

Which would be incorporated into the agent statefulset template, to include the limits spec:

...
resources:
  limits:
    gpu-vendor.example/example-gpu: 1

Then in my custom python processor, I can do image manipulations and bring in my own models to do fun things :)

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions