PyTorch Kubeflow Pipeline Components

PyTorch Kubeflow Pipeline Components提供了一套SDK和组件，使您能够使用PyTorch构建kubeflow流水线。您可以使用此存储库中预定义的组件，使用Kubeflow Pipelines SDK构建您的流水线。

安装

需求

Python >= 3.6 Kubeflow集群设置（本地或任何云中）

安装最新版本

使用以下命令从PyPI安装PyTorch Pipeline Components。

pip install -U pytorch-kfp-components

从源安装

使用以下命令从GitHub安装PyTorch Kubeflow Pipeline Components。

git clone https://github.com/kubeflow/pipelines.git
pip install pipelines/components/PyTorch/pytorch_kfp_components/.

运行测试

运行以下命令

pip install tox
cd ./components/PyTorch/pytorch-kfp-components/
tox -rvve py38

示例

要运行示例，请按照以下说明操作

示例 README.md