Support GPU in Kubernetes cluster
ddieruf opened this issue · 0 comments
ddieruf commented
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 :)