自动化与运维 ✓ NVIDIA · 官方

TAO Kubernetes 作业运行

通过 Kubernetes 提交和监控带 NVIDIA GPU 的 TAO 训练作业。

53/ 100
谨慎使用

价值存在,但可靠性、证据或控制措施仍有明显缺口。

查看评分依据 ↓
可直接使用于
Codex · Claude Code
Star 数
★ 3.8k
最近更新
今天
License
Apache-2.0
kubernetesgpu-schedulingtao-toolkitdistributed-training
+2container-jobsnvidia-gpu-operator

这个 Skill 能做什么,适合哪些场景?

该技能使用 NVIDIA TAO SDK 将 TAO 容器提交为 Kubernetes Job,默认运行单 Pod GPU 作业。它支持通过 kubeconfig 或 Pod 内服务账户连接 EKS、GKE、AKS 和本地集群,并可通过 Indexed Job 进行多节点分布式训练。技能还提供作业状态、日志、副本诊断、失败分析、取消和自动清理指导。它要求 GPU 节点运行时、TAO SDK、可认证的集群以及 NVIDIA GPU Operator 或设备插件已就绪。

执行 GPU 节点运行时、Python SDK、集群认证和 GPU 资源的预检;调用 KubernetesSDK.create_job() 创建带 GPU 资源限制的 Kubernetes Job;传递环境变量、S3 输入输出、镜像拉取密钥和节点选择器;通过 get_job_status()、get_job_logs()、get_job_replicas() 和 get_failure_analysis() 读取状态、日志和故障信息;调用 cancel_job() 删除运行中的作业;在多节点模式下创建无头 Service 和 Indexed Job,并注入 TAO 或 torchrun 所需的 rendezvous 环境变量。

适合
  • 需要在 EKS、GKE、AKS 或本地 GPU 集群中运行 TAO 训练的 ML 平台工程师。
  • 希望把 TAO 容器接入现有 Kubernetes 原生平台、通过 Python SDK 提交作业的团队。
  • 需要使用 S3 数据集输入和结果输出,并监控 GPU 作业日志与失败原因的工程师。
  • 需要在多节点、多 GPU 环境运行 PyTorch 分布式 TAO 训练的用户。

如何安装这个 Skill?

使用前请注意
  • 执行会访问并改变 Kubernetes 集群状态;在提交或取消 Job 前应确认目标集群、命名空间、RBAC、GPU 配额、镜像来源和数据路径。
  • 不要将 NGC、S3 或 HF 凭据直接暴露在日志、命令历史或 Job 环境中;文档未充分说明凭据脱敏和 Pod 内最小权限。
  • 多节点 Indexed Job 不具备 gang scheduling 或弹性恢复能力,部分 Pod 调度成功时可能导致 rank-0 长时间等待。
  • 预检依赖未提供的共享技能、脚本和版本文件,使用前应核对这些路径与 SDK 版本;中国大陆网络环境还需验证 PyPI、nvcr.io 和 Helm 可达性。
开始前你需要
Agent 需要具备
  • Shell / 命令行
  • 网络访问
  • 本地文件系统
需要预先安装
  • Python
  • nvidia-tao-sdk[kubernetes]
  • Kubernetes cluster
  • NVIDIA driver 580
  • CUDA Toolkit 13.0
  • NVIDIA Container Toolkit 1.19.0
  • NVIDIA GPU Operator or device plugin
  • kubeconfig or in-cluster service account
  • kubectl for optional diagnostics
  • Helm for GPU Operator installation

使用仓库 README 支持的 skills CLI 安装指定技能:

npx skills add nvidia/skills --skill tao-run-on-kubernetes --yes

安装后,技能会在代理下一次加载技能并遇到相关任务时可用。运行前还需安装 Python 依赖:python -m pip install 'nvidia-tao-sdk[kubernetes]'。源材料未规定其他客户端的专用安装步骤。

通用方式:手动安装到 Claude Code(macOS / Linux)
tmp="$(mktemp -d)"
git clone --depth 1 https://github.com/NVIDIA/skills.git "$tmp"
mkdir -p ~/.claude/skills
cp -R "$tmp/skills/tao-run-on-kubernetes" ~/.claude/skills/
rm -rf "$tmp"

根据源仓库地址和 Skill 路径自动生成,只复制这个 Skill 的文件夹。如果上文有作者提供的安装方式,请优先按作者说明操作;想只在当前项目中使用,把 ~/.claude/skills 换成项目里的 .claude/skills。

如何使用这个 Skill?

试试这样说

安装后,把下面任意一句发给 Agent 即可触发:

  • 在已有 NVIDIA GPU Operator 的 EKS 集群中,通过 TAO SDK 提交一个使用 1 张 GPU 的 DINO 训练 Job,并监控日志。

向代理提出类似请求:“在已有 NVIDIA GPU Operator 的 EKS 集群中,通过 TAO SDK 提交一个使用 1 张 GPU 的 DINO 训练 Job,并监控日志。”

技能会先检查 GPU 节点运行时、TAO SDK、集群认证和 GPU 资源,然后使用 KubernetesSDK.create_job() 提交作业。多节点训练可在 create_job() 中设置 num_nodes > 1;S3 工作流需要相应的 S3 凭据。

这个 Skill 有哪些优点和局限?

优点
  • 使用 Kubernetes Job 和 NVIDIA GPU 资源限制,适合接入现有云端或本地集群。
  • 提供预检、状态、日志、副本诊断、失败分析、取消和 TTL 自动清理流程。
  • 支持单节点和 Indexed Job 多节点分布式训练,并同时兼容 TAO entrypoint 与原生 torchrun 环境变量。
  • 明确区分自管 GPU 节点与云厂商托管节点的运行时检查方式。
局限
  • 依赖 NVIDIA driver 580、CUDA Toolkit 13.0、Container Toolkit 1.19.0、TAO SDK 和 GPU Operator 或设备插件。
  • 不支持弹性或容错训练;Indexed Job 的 backoff_limit 为 0。
  • 不提供 gang scheduling、MPI/Horovod 或 PVC 共享存储支持。
  • 多节点作业要求 Kubernetes 1.28+、节点间 29500 端口连通,并且源材料未提供测试套件或平台验证矩阵。

这个 Skill 与同类方案有什么区别?

与相关 Skills 并排比较;分数均按同一 FSRS 标准得出。

Skill FS 评分 Star 数 最近更新 License
TAO Kubernetes 作业运行 本页 ✓ NVIDIA · 官方 53 · 谨慎使用 ★ 3.8k 今天 Apache-2.0
TAO 推理微服务部署技能 ✓ NVIDIA · 官方 45 · 不推荐 ★ 3.8k 今天 Apache-2.0
Dynamo 部署故障排查 ✓ NVIDIA · 官方 52 · 谨慎使用 ★ 3.8k 今天 Apache-2.0
TAO NVIDIA GPU 主机配置 ✓ NVIDIA · 官方 55 · 谨慎使用 ★ 3.8k 今天 Apache-2.0
TAO 本地 Docker 运行 ✓ NVIDIA · 官方 48 · 谨慎使用 ★ 3.8k 今天 Apache-2.0

技能明确建议:需要弹性 PyTorch 训练时使用 Kubeflow Training Operator;需要 MPI 或 Horovod 时使用 MPI Operator;需要 gang scheduling、队列或配额时使用 Volcano 或 Kueue。该技能的 Indexed Job 路径更简单且依赖较少。

FollowSkills 如何评估这个 Skill?

FollowSkills 评估 · FSRS-2.0
谨慎使用
53/ 100 五分制 2.7 / 5
上游仓库在本次评估后已有新提交;当前评分仍对应所示审查版本,可能尚未覆盖最新改动。
1信任安全16 / 25 · 3.2/5

文档说明了 kubeconfig、服务账户、命名空间、S3、NGC 和模型凭据的数据流,并要求在节点安装前取得用户批准;但 allowed-tools 含 Bash,提交 Job、取消 Job、安装依赖和删除资源会产生外部影响,缺少通用的逐次确认、最小 RBAC、敏感凭据脱敏和恢复流程,因此扣分。

2可靠稳定8 / 20 · 2.0/5

预检、状态、日志、副本诊断、失败分类和 backoff_limit=0 使主路径及异常反馈较清晰;但依赖未提供的 tao-run-platform、versions.yaml、脚本路径和 SDK 实现,TAO_SKILL_BANK_ROOT 与 TAO_SKILL_BANK_PATH 变量还存在不一致,且无本技能专属测试覆盖,因此按静态上限保守扣分。

3适用触发10 / 15 · 3.3/5

目标用户、支持的 EKS/GKE/AKS/自建集群、单节点与多节点场景以及不支持弹性训练、gang scheduling、MPI/Horovod 和 PVC 的边界较明确;但缺少语义触发规则、输入输出契约和中文支持说明,且依赖 PyPI、nvcr.io、Helm 和海外文档,部分中国大陆网络环境可能受限,因此扣分。

4规范维护9 / 15 · 3.0/5

文档包含前置检查、配置、API、监控、清理、分布式训练、替代方案和常见错误,且有版本、Apache-2.0 声明、skill card 和评估报告;但缺少推荐的 Instructions/Examples 章节,作者格式不合规范,版本变更、维护责任和更新路径不清晰,许可证元数据为 NOASSERTION,配置与 SKILL.md 也有细节不一致,因此扣分。

5有效结果6 / 15 · 2.0/5

提供了可直接改写的 Python create_job、监控、日志、失败分析和取消示例,覆盖 GPU Job 的核心任务,且明确推荐 operator 替代复杂场景;但示例依赖未随证据提供的 SDK/共享技能和外部集群,唯一评估任务主要测试文档总结而非实际提交结果,直接可用性和相对手工操作的收益证据有限,因此不超过静态上限并扣分。

6证据核验4 / 10 · 2.0/5

证据包含固定 revision、结构化配置、评估任务、结果摘要、参考文档名称及基准日期;但只有一个正向任务、没有负向或关键路径复现材料,基准结果缺乏可核验日志或提交的测试套件,无法独立验证主要运行时声明,因此扣分。

1 2 3 4 5 6

点击维度查看打分理由

评估于 2026年7月20日 审查版本 55f18499943e 评估证据[1][2][3][4][5][6][7][8][9][10][11][12][13]

证据充分度:低 — 主要依赖静态检查、作者材料或有限演示;适合发现线索,不适合做高风险决策。

查看完整评分方法 →

常见问题

它是否支持只使用 kubectl 提交作业?
不支持。该技能是 SDK-only,要求通过 nvidia-tao-sdk 的 KubernetesSDK 提交作业。
我需要哪些 Kubernetes 权限和凭据?
需要已认证的 kubeconfig 或 Pod 内服务账户,并确认目标命名空间可以创建 Job。使用 nvcr.io 镜像时,还需预先创建镜像拉取 Secret。
多节点训练失败时会自动重启吗?
不会。Indexed Job 使用 backoff_limit=0;弹性或容错训练应使用 Kubeflow 的 PyTorchJob operator。
作业完成后会保留多久?
默认 ttl_seconds_after_finished=3600,终止状态后的 Job 会在约一小时后自动删除。

同仓库的其他 Skills

均来自 NVIDIA/skills

自动化与运维 ✓ NVIDIA · 官方

TAO 推理微服务部署技能

在多种计算平台上启动、调用并停止 TAO 推理微服务。

★ 3.8k FS 45 不推荐 今天
自动化与运维 ✓ NVIDIA · 官方

Dynamo 部署故障排查

按证据定位不健康的 Dynamo Kubernetes 部署。

★ 3.8k FS 52 谨慎使用 今天
自动化与运维 ✓ NVIDIA · 官方

TAO NVIDIA GPU 主机配置

为 TAO 的 Docker 与 Kubernetes GPU 主机检查并配置 NVIDIA 驱动、CUDA 和容器运行时。

★ 3.8k FS 55 谨慎使用 今天
开发与工程 ✓ NVIDIA · 官方

TAO Sparse4D 训练技能

帮助代理训练、评估、导出、量化和推理多摄像头时序3D检测跟踪模型。

★ 3.8k FS 54 谨慎使用 今天
自动化与运维 ✓ NVIDIA · 官方

TAO 本地 Docker 运行

帮助 AI 代理在本机或远程单节点 Docker GPU 主机上运行 TAO 作业。

★ 3.8k FS 48 谨慎使用 今天
开发与工程 ✓ NVIDIA · 官方

TAO 平台执行 SDK

在多种 GPU 平台上提交、监控并管理 NVIDIA TAO 训练任务。

★ 3.8k FS 55 谨慎使用 今天
数据与分析 ✓ NVIDIA · 官方

TAO 标准训练流程

为 TAO 模型执行标准的训练、评估与导出流程。

★ 3.8k FS 52 谨慎使用 今天
数据与分析 ✓ NVIDIA · 官方

TAO BEVFusion 训练技能

用 TAO 训练、评估和推理融合激光雷达与摄像头数据的 BEV 三维检测模型。

★ 3.8k FS 49 谨慎使用 今天
开发与工程 ✓ NVIDIA · 官方

NeMo MBridge GPU 内存调优

诊断 Megatron Bridge 训练中的 GPU OOM,并降低碎片、激活和 PEFT 内存占用。

★ 3.8k FS 61 推荐 今天
自动化与运维 ✓ NVIDIA · 官方

TAO 工作流启动门禁

在执行 TAO 作业前完成平台、凭证、数据与容器预检。

★ 3.8k FS 60 推荐 今天
开发与工程 ✓ NVIDIA · 官方

Megatron Bridge 激活重计算

在 Megatron Bridge 中用额外计算换取更低的 GPU 激活内存占用。

★ 3.8k FS 59 推荐 今天
自动化与运维 ✓ NVIDIA · 官方

Megatron-LM 集群训练启动助手

帮助你在 SLURM GPU 集群上可靠启动、监控和排查多节点 Megatron-LM 训练任务。

★ 3.8k FS 56 谨慎使用 今天
开发与工程 ✓ NVIDIA · 官方

NeMo MBridge CPU Offload

为 Megatron Bridge 配置、验证并排查 CPU 卸载,以缓解 GPU 显存压力。

★ 3.8k FS 55 谨慎使用 今天
开发与工程 ✓ NVIDIA · 官方

NeMo MBridge CUDA 图优化

在 Megatron Bridge 训练中配置、验证并评估 CUDA Graph,以降低主机驱动开销。

★ 3.8k FS 55 谨慎使用 今天
自动化与运维 ✓ NVIDIA · 官方

TAO SLURM 集群运行

通过 SSH、SLURM 和 Pyxis/Enroot 在远程 GPU 集群上运行 TAO 作业。

★ 3.8k FS 55 谨慎使用 今天
开发与工程 ✓ NVIDIA · 官方

NeMo MBridge 配方推荐器

根据模型、GPU规模和训练目标推荐可调整的Megatron Bridge配方。

★ 3.8k FS 53 谨慎使用 今天
数据与分析 ✓ NVIDIA · 官方

OCDNet 场景文本检测

用 TAO Toolkit 训练和部署任意方向场景文本检测模型。

★ 3.8k FS 53 谨慎使用 今天
自动化与运维 ✓ NVIDIA · 官方

TAO Brev GPU 运行技能

在 Brev GPU 实例上运行和管理 TAO 训练、评估与推理任务。

★ 3.8k FS 51 谨慎使用 今天
自动化与运维 ✓ NVIDIA · 官方

Dynamo 配方部署助手

在 Kubernetes 上选择、校验、修补并部署现有的 NVIDIA Dynamo 推理配方。

★ 3.8k FS 50 谨慎使用 今天
数据与分析 ✓ NVIDIA · 官方

TAO MAL 弱监督分割

用点或框标注训练模型,生成弱监督语义分割掩码。

★ 3.8k FS 50 谨慎使用 今天

相关 Skills