如何解决如何在顶点 AI 管道中指定机器类型
我想使用 kfp sdk 在顶点 ai 管道中指定机器类型。 我不知道如何在将其作为管道组件执行时指定 machine_type。 我尝试了 kfp.v2.google.experimental.run_as_aiplatform_custom_job,但它作为 CustomJobExecution 而不是 ContainerExecution 运行。 为此,我想使用Airtifact,但该组件上未安装airtifact。
由于想使用之前的组件的airtifact和Output[Airtifact]的功能,所以想把它作为ContainerExecution而不是CustomJobExecution来执行。
解决方法
您可以使用 set_memory_limit 和 set_cpu_limit,就像使用 Kubeflow Pipelines 一样。 Vertex Pipelines 会将这些限制转换为满足您要求的机器类型。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。