AI 资源库 · 已整理 3 项

GPU 调度

查看与GPU 调度相关的 AI 资讯、工具、Skill、工作流和开源项目。

Nebius 将 rollout 收集时间缩短 66.9%——改变了什么

Nebius 的 AI 训练团队在 Terminal-Bench 类负载上做了一次实验:把一次 256 条轨迹的批次收集时间减少了 66.9%。原来在智能体强化学习训练中,模型要先完成任务、拿到结果才能学习,但任务长短不一,慢的几条会让整批训练卡住,昂贵的 GPU 常常在等便宜的沙箱和代码执行。做法是把推理引擎当成训练优化器的一部分:按整组完成而不是启动顺序来组批,让生成和训练部分重叠、缓存对话历史避免重复预填,并把基础设施故障和模型自身失败分开计分。这只是一个实验中的批次收集测量,不含训练器本身的前向反向计算,不等于完整训练提速。对读者来说,这提供了一个思路:智能体训练慢,瓶颈往往不在 GPU 算力,而在任务执行和调度这些配套环节。

文中可用工具

  • Nebius提供 AI 训练与推理服务的公司,本文实验由其实施
  • Terminal-Bench用于评测终端操作类智能体任务的基准,本文实验基于它构造负载
  • GRPO一种强化学习训练方法,把同一任务的多次尝试相互比较打分
☆ 收藏查看来源 ↗

在扩容 GPU 池之前,先衡量实际产出

这篇文章讲的是在 Kubernetes 上跑 AI 任务的团队,常看到 GPU 全被占满、批处理队列越排越长,第一反应就是加 GPU。但作者提醒,GPU 被分配、在忙、和真正产出有价值的结果是三件不同的事,队列变长也可能是时机、服务容量或模型和硬件选型的问题。解决思路是先想清楚每个任务的截止时间和产出标准,再用错峰排队把现有四块 GPU 在两小时内跑完原本需要八块的四个任务,同时用接近真实负载的压测摸清可持续吞吐,并把 OpenCost 的成本数据和实际产出对上,算出每个合格结果花多少钱。这样读者能在花钱扩容之前,先确认现有容量是否已用足、瓶颈到底在哪。Kueue、vLLM、DistServe、OpenCost 等工具文中都有提到,具体接入方式原文未详细说明。

文中可用工具

  • Kueue做准入控制,按配额和策略决定任务何时被放行
  • vLLM模型服务框架,其 KV-cache 占用率可反映缓存使用情况
  • DistServe说明评估服务容量时应以延迟等应用层目标为准
  • OpenCost按时间间隔统计工作负载成本,并可追踪每百万 token 的推理成本
☆ 收藏查看来源 ↗

AI 任务排队等候,GPU 却闲置:Kubernetes 有了新方案

这篇文章是 KubeCon 前瞻专栏,讲 Kubernetes 社区怎么让 GPU 更好地服务 AI。问题在于:负责把任务分配到现有机器的调度器和负责加机器的自动扩缩器各干各的,结果任务卡在排队,昂贵的 GPU 却闲着。目前有几个方向:HPE 用开源工具自动收集 Pod 重启信息并推送到 Slack,减少人工排查;AWS 提出让扩缩器更懂调度器,并用动态资源分配、工作负载感知调度和 provisioning requests 衔接两边;SIG-Autoscaling 已有公开提案,让两个环节直接对话。另外 CNCF 沙箱项目 llm-d 发布 v1.0,IBM 测试中智能体任务推理成本降低 11.5 倍、缓存复用约 98%。这些多数是提案或演示,尚未说明何时默认可用。

文中可用工具

  • HPE Private Cloud AIHPE 的私有云 AI 平台,用于托管长时间运行的 AI/ML 任务
  • k8s-pod-restart-info-collector开源 Kubernetes 排障工具,自动收集 Pod 重启信息并发送到 Slack
  • llm-dCNCF 沙箱项目,用于在云原生环境分布式部署 AI 推理,已发布 v1.0
  • dynamic resource allocation (DRA)Kubernetes 的动态资源分配机制,让扩缩器更了解 GPU 等资源需求
  • provisioning requests连接调度器和扩缩器的配套机制,原文未说明具体用法
☆ 收藏查看来源 ↗