Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Installation for kubeflow pipeline v2 GPU node pool (function "add_node_selector_constraint") #3616

Open
yonggunpeak opened this issue Nov 1, 2023 · 2 comments
Open

Installation for kubeflow pipeline v2 GPU node pool (function "add_node_selector_constraint") #3616

yonggunpeak opened this issue Nov 1, 2023 · 2 comments
Labels
area/pipelines AREA: Kubeflow Pipelines

Comments

@yonggunpeak
Copy link

import kfp.dsl as dsl
gpu_op = dsl.ContainerOp(name='gpu-op', ...).set_gpu_limit(2)
gpu_op.add_node_selector_constraint('cloud.google.com/gke-accelerator', 'nvidia-tesla-p4')

It looks like it doesn't work on v2 version.
Is there any guide for v2 version? If so or not, can you give me guide how I should do?
Thank you
@varodrig
Copy link
Contributor

@kubeflow/wg-pipeline-leads

@varodrig
Copy link
Contributor

/area pipelines

@google-oss-prow google-oss-prow bot added the area/pipelines AREA: Kubeflow Pipelines label Jan 25, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
area/pipelines AREA: Kubeflow Pipelines
Projects
Needs Triage
Status: To Do
Milestone
No milestone
Development

No branches or pull requests
2 participants
@varodrig @yonggunpeak