入门
创建你的第一个流水线
要开始学习教程,请使用 pip 安装 kfp
v2
pip install kfp
这是一个打印问候语的简单流水线
from kfp import dsl
@dsl.component
def say_hello(name: str) -> str:
hello_text = f'Hello, {name}!'
print(hello_text)
return hello_text
@dsl.pipeline
def hello_pipeline(recipient: str) -> str:
hello_task = say_hello(name=recipient)
return hello_task.output
你可以使用 KFP SDK DSL Compiler
将流水线编译为 YAML
from kfp import compiler
compiler.Compiler().compile(hello_pipeline, 'pipeline.yaml')
dsl.component
和 dsl.pipeline
装饰器将你的类型注解 Python 函数转换为组件和流水线。KFP SDK 编译器将领域特定语言 (DSL) 对象编译成一个自包含的流水线YAML 文件。
你可以将 YAML 文件提交到兼容 KFP 的后端执行。如果你已经部署了KFP 开源后端实例并获得了部署的端点,可以使用 KFP SDK Client
提交流水线进行执行。以下示例使用参数 recipient='World'
提交流水线执行
from kfp.client import Client
client = Client(host='<MY-KFP-ENDPOINT>')
run = client.create_run_from_pipeline_package(
'pipeline.yaml',
arguments={
'recipient': 'World',
},
)
客户端将打印一个链接,用于在 UI 中查看流水线执行图和日志。在此示例中,流水线有一个任务,负责打印并返回 'Hello, World!'
。
后续步骤
在接下来的几节中,你将了解有关创作流水线的核心概念以及如何创建更具表现力、更有用的流水线。
最后修改时间:2024年7月6日: Partial Revert of #3787 (#3794) (b0d73bc)