kubeflow - 如何在顶点ai管道中指定机器类型

Question

我想使用 kfp sdk 在顶点 ai 管道中指定机器类型。我不知道如何在将 machine_type 作为管道组件执行时指定它。我尝试了 kfp.v2.google.experimental.run_as_aiplatform_custom_job，但它作为 CustomJobExecution 而不是 ContainerExecution 运行。出于这个原因，我想使用 Airtifact，但是这个组件上没有挂载 airtifact。

由于我想使用前面组件的airtifact和Output [Airtifact]的功能，所以我想将它作为ContainerExecution而不是CustomJobExecution来执行。

Answer 1

您可以使用 set_memory_limit 和 set_cpu_limit，就像使用 Kubeflow Pipelines 一样。Vertex Pipelines 会将这些限制转换为满足您要求的机器类型。