基于 Kubernetes 的 Kubeflow 平台安装清单,包含 Pipelines、KServe、Katib 和中央仪表盘等组件。
kubeflow/katib
基于 Kubernetes 的自动化机器学习系统,支持超参数调优、早停与神经架构搜索。
- Star
- 1,707
- Fork
- 550
- 未关闭 Issue
- 120
- 最近提交
- 7天前
这是什么?
Katib 是一个面向 Kubernetes 的自动化机器学习(AutoML)项目,提供超参数调优、早停和神经架构搜索功能。它与具体的机器学习框架无关,只要应用能作为 Kubernetes 工作负载运行并上报指标,任何语言编写的训练程序都可以进行调优。
试验(Trial)通过 Kubernetes 自定义资源启动,开箱支持 Kubeflow Training Operator、Argo Workflows 和 Tekton Pipelines。搜索算法包括随机搜索、网格搜索、贝叶斯优化、TPE、CMA-ES、HyperBand 和基于种群的训练(PBT)等,并可借助 Goptuna、Hyperopt、Optuna 和 Scikit-Optimize 等库实现。
用户既可以通过 kubectl 将 Katib 控制平面独立部署到集群中,也可以使用 Python SDK(kubeflow-katib)在 Python 中定义并提交调优实验。
核心功能
- 01支持超参数调优、早停和神经架构搜索。
- 02兼容任何可作为 Kubernetes 自定义资源运行的训练程序,与框架无关。
- 03内置贝叶斯优化、TPE、CMA-ES、HyperBand、PBT 等搜索算法。
- 04可与 Kubeflow Training Operator、Argo Workflows、Tekton Pipelines 集成。
- 05提供 Python SDK,便于创建和提交调优实验。
适用场景
- 在 Kubernetes 集群上为 TensorFlow、PyTorch、XGBoost 等模型调优超参数。
- 将大规模并行搜索任务分布到集群资源上运行,而非单台机器。
- 使用 ENAS 或 DARTS 搜索神经网络架构。
- 为长时间运行的训练任务加入自动早停。
快速上手
- 1
在 Kubernetes 集群上安装 Katib 控制平面(稳定版 v0.17.0)
kubectl apply -k "github.com/kubeflow/katib.git/manifests/v1beta1/installs/katib-standalone?ref=v0.17.0" - 2
从 PyPI 安装 Katib Python SDK
pip install -U kubeflow-katib - 3
按照入门指南,使用 Python SDK 创建第一个超参数调优实验
使用前须知
- 需要一个正在运行的 Kubernetes 集群,并且需要具备 Kubernetes 知识才能安装和使用。
- 安装前提条件和配置细节仅在外部 Kubeflow 文档中说明。
标签
常见问题
- katib 是什么?
- 基于 Kubernetes 的自动化机器学习系统,支持超参数调优、早停与神经架构搜索。 Katib 是一个面向 Kubernetes 的自动化机器学习(AutoML)项目,提供超参数调优、早停和神经架构搜索功能。它与具体的机器学习框架无关,只要应用能作为 Kubernetes 工作负载运行并上报指标,任何语言编写的训练程序都可以进行调优。
- katib 可以免费使用吗?用的什么许可证?
- katib 以 Apache-2.0 许可证开源。用于商业产品前,请先确认许可证条款。
- katib 还在维护吗?
- 活跃维护中。katib 最近一次提交是在 2026年10月31日。
- katib 适合用来做什么?
- 在 Kubernetes 集群上为 TensorFlow、PyTorch、XGBoost 等模型调优超参数。 将大规模并行搜索任务分布到集群资源上运行,而非单台机器。 使用 ENAS 或 DARTS 搜索神经网络架构。 为长时间运行的训练任务加入自动早停。
- katib 怎么安装和运行?
- 在 Kubernetes 集群上安装 Katib 控制平面(稳定版 v0.17.0): kubectl apply -k "github.com/kubeflow/katib.git/manifests/v1beta1/installs/katib-standalone?ref=v0.17.0" 从 PyPI 安装 Katib Python SDK: pip install -U kubeflow-katib 按照入门指南,使用 Python SDK 创建第一个超参数调优实验
- 在哪里获取 katib?
- 源代码和安装说明在 GitHub 上:https://github.com/kubeflow/katib
概要由 AI 根据项目 README 生成;GitHub 数据每日更新。
相关项目
用于跟踪、可视化和管理机器学习实验、数据集与模型的 Python 库及托管平台。
通过自然语言描述需求和表格数据,由多智能体流程自动构建机器学习模型。
用于在 Amazon SageMaker 上训练和部署机器学习模型的 Python SDK。
基于 Google Cloud Vertex AI 的 Notebook、代码示例与示例应用,演示机器学习和生成式 AI 工作流。
一个 Python 框架,用于构建、扩展和部署 AI/ML 工作流,从 notebook 原型到生产编排。