PyTorch Kubeflow Pipeline Components provides an SDK and a set of components that lets you build kubeflow pipelines using PyTorch. You can use the predefined components in this repository to build your pipeline using the Kubeflow Pipelines SDK.
Python >= 3.6 Kubeflow cluster setuo (on-prem or in any of the Clouds)
Use the following command to install PyTorch Pipeline Components from PyPI.
pip install -U pytorch-kfp-components
Use the following commands to install PyTorch Kubeflow Pipeline Components from GitHub.
git clone https://github.com/kubeflow/pipelines.git
pip install pipelines/components/PyTorch/pytorch_kfp_components/.
Run the following command
pip install tox
cd ./components/PyTorch/pytorch-kfp-components/
tox -rvve py38
For running the samples follow the instruction mentioned as below