Как указать тип машины в конвейере Vertex AI

Question

Я хочу указать тип машины в конвейере вершины ai с использованием kfp sdk. Я не знаю, как указать machine_type при выполнении его как компонента конвейера. Я пробовал kfp.v2.google.experimental.run_as_aiplatform_custom_job, но он работал как CustomJobExecution вместо ContainerExecution. По этой причине я хочу использовать Airtifact, но на этом компоненте не установлен airtifact.

Поскольку я хочу использовать воздушный эффект предыдущих компонентов и функцию Output [Airtifact], я хочу выполнить его как ContainerExecution вместо CustomJobExecution.

Answer 1

Вы можете использовать set_memory_limit и set_cpu_limit, как если бы вы использовали Kubeflow Pipelines. Vertex Pipelines преобразует эти ограничения в тип машины, удовлетворяющий вашему запросу.