top of page

AWS Batch 配额管理让团队共享 GPU 并回收其分配额度

1天前
讀畢需時 12 分鐘

AWS Batch 配额管理让团队可以出借闲置训练容量,同时保留从借用工作负载中回收容量的能力。该功能将资源分配与 Amazon SageMaker Training 的调度决策关联起来;后者是 AWS 用于训练机器学习模型的托管服务。其核心取舍很直接:团队可以使用其他人的闲置容量,但其任务必须能够承受资源所有者收回容量时失去访问权限的情况。

AWS 在 10 月 29 日的演示中,通过不同的工作负载策略说明了这一取舍。根据该公司的配额管理公告,这项功能本身于 2026 年 3 月 25 日推出。两者的区别很重要:本周的动态是对既有功能的详细运营示例,而非其首次发布。

对于基础设施团队而言,竞争方案并不陌生:通过严格隔离保护分配额度,或汇集资源,并在需求冲突时协商访问权限。AWS 将这些协商纳入调度策略。这将压力转向平台管理员和训练负责人,他们必须决定哪些实验应获得可预测的访问权限,哪些可以接受中断。

AWS Batch 配额管理将分配额度转化为调度规则

这一变化在于:将团队的分配额度与其收回被其他团队借用容量的权利关联起来。

配额份额是一种具有容量限制和共享规则的虚拟队列,代表一个团队、项目或工作负载分组。管理员将这些份额附加到作业队列,即承载已提交工作的容器。AWS 文档说明,每个队列最多可包含 20 个份额,每个份额最多可设置五项容量限制。

这些限制使用受支持的训练实例类型,即特定的机器配置,而不是一个由可互换 GPU 构成的无差别资源池。AWS 的配额份额策略提供三种选择。RESERVE 为自身份额保留闲置容量。LEND 允许其他方使用闲置容量。LEND_AND_BORROW 则允许出借和借用,但须受配置的借用上限约束。

这是一个对临时访问设有明确规则的分配系统。团队的应得额度与其当前消耗不再需要在每一刻完全一致。闲置容量可以服务于其他工作,而不会永久改变底层分配。

这种区分让共享基础设施更容易理解。分配额度回答的是:当需求回归时,谁拥有主张权。消耗回答的是:谁正在使用资源。将两者视为不同问题,使管理员能够在不放弃所有权的前提下提供灵活性。

设想一个假设中的平台团队,在多个研究小组之间分配训练访问权限。固定分区可以保护每个小组免受竞争影响,却可能让资源闲置。无限制资源池可以在平静时段改善访问体验,但会让平台团队不得不解释为何已分配资源不可用。

配额策略提供了一种中间立场。借用成为带有既定回收机制的条件式访问。该安排仍需要就分配额度达成一致,但管理员可以在调度争议发生前将协议编码化。

AWS 将该功能用于连接到 SAGEMAKER_TRAINING 服务环境的 SageMaker Training 队列;该环境是与托管训练执行相关的 Batch 配置。读者不应将其理解为适用于所有 AWS Batch 工作负载的通用资源共享功能。

在评估迁移时,这一范围很重要。已经提交 SageMaker 作业的团队,与通过其他执行系统管理训练的团队,面临的运营决策不同。调度策略应置于更大的工作流程中考虑。

因此,实际问题比是否提供共享更具体:组织能否通过这些份额表达现有承诺,同时不对有保障的访问权限制造误导性预期。

平台团队必须界定分配额度究竟承诺什么

只有在团队就借用容量附带的条件达成一致后,自动出借才能减少调度协调。

直接压力落在负责共享训练基础设施的平台团队身上。他们的职责不再只是分配访问权限,还必须区分长期分配额度与临时使用闲置资源的许可。

这种区分在期限临近前听起来像行政事务。借用团队可能将一次成功的训练运行视为容量可用的证据。出借团队则可能将同一批资源视为其预期收回的分配额度的一部分。

两种理解都可以理解。没有明确策略时,分歧会变得个人化:必须有人决定停止谁的实验。有了明确策略,分歧会更早地进入分配协议环节。

这正是 GPU 容量共享成为组织问题的原因。调度器可以一致地应用规则,但一致性并不能证明规则符合业务优先级。管理员仍需要为每个份额的处理方式提供站得住脚的解释。

研究工作提供了一个有用的假设性对比。探索性实验通常有灵活的完成窗口,而与评估截止日期挂钩的训练运行则较少有等待空间。这些差异可以证明不同共享策略的合理性,并不意味着某个团队的工作缺乏价值。

困难的情况发生在团队内部。一个小组可能同时负责探索性实验、发布准备和紧急调查。为整个小组分配单一策略,可能会把中断成本不同的工作负载当作相同对待。

按项目或工作负载组织份额,有助于表达这些差异。但这也会产生更多需要维护的策略边界。正确的分组方式取决于用户能否在提交前可靠地对工作进行分类。

优先级带来另一个治理问题。如果用户每逢期限临近便提高作业紧急度,优先级数值可能失去意义。每个团队都有动机保护自己的完成时间。

应对措施应在运营层面具体明确。团队需要共同的紧急度定义、例外情况的负责人,以及审查变更的机制。否则,自动化调度可能会通过配置变更重现同样的协商过程。

例如,一个组织可以区分错过实验目标与阻碍发布决策。这是一项示例性策略选择,而非 AWS 内置的分类。它的价值在于,在容量变得紧张之前让后果变得可以理解。

文档能够支持这种共识。工程团队可以在一个可搜索的知识库中记录分配决策、回收假设和例外情况。有价值的记录应解释策略存在的原因,以及谁可以修改它。

当调度器接手执行工作后,这项工作并不会消失,反而会更加显性化。此前存在于消息或会议中的借用协议,现在必须经受实际工作负载行为的检验。

这也是为什么推广应从工作负载特征而非仅仅从团队名称开始。中断容忍度、恢复行为和对截止日期的敏感性,比组织架构图更能为策略提供可靠基础。

借用容量附带中断约定

AWS 的示例将中断容忍度视为选择共享策略的理由,而非事后考虑。

10 月的演示配置了三个份额,每个份额分配一个 ml.g5.xlarge 实例。其后台工作份额可以出借和借用,并启用了基于优先级的中断。AWS 明确说明,这些作业使用检查点,即保存的训练状态,可支持在中断后恢复。

中优先级份额会出借闲置容量,但不借用。其作业会在自身分配额度繁忙时等待。受保护份额既不出借也不借用,并禁用了其自身作业之间的中断。

在SageMaker 配额示例中,出借份额中的新作业会导致 Batch 从借用份额回收容量。另一个情况是,提高等待作业的优先级会使其份额内的另一项作业让出资源。

这是两种不同机制。跨份额抢占,即停止工作以将容量恢复给其分配份额,执行的是出借协议。份额内抢占,即停止同一份额内优先级较低的工作,执行的是该份额内部的紧急度规则。

该示例表明,不同策略可以在同一总体安排中并存。它并未证明每种工作负载都能从最宽松的借用配置中受益。

对于后台实验,借用容量提供了更早取得进展的机会。其负责人接受完成时间的不确定性。对于受截止日期约束的任务,避免这种不确定性可能比在平静时段获得额外容量更有价值。

这种区别改变了团队应如何描述借用。它是可能发生中断的临时访问,而非团队可靠分配额度的延伸。

中断约定应回答几个实际问题。工作负载能否恢复?谁负责处理恢复失败?预期完成时间是否假定持续访问?同一作业反复让出资源时会发生什么?

即使系统会处理停止和重新提交工作,这些问题仍然重要。自动化控制执行状态转换。应用负责人仍需决定由此产生的行为是否可接受。

这还涉及沟通问题。一个开始运行的作业会带来进展预期。用户需要理解,在借用容量上运行,并不具备在自身分配额度内运行所具有的同等连续性预期。

否则,成功借用本身可能造成采纳问题。团队习惯了闲置容量,将其纳入交付假设,随后又将合理的回收视为服务故障。

因此,策略需要出现在规划语言中,而不仅仅是配置里。预测应区分由已分配容量支撑的工作,与通过机会性访问加速的工作。

这并不会降低借用执行的价值,反而使其价值更容易评估。只要团队不据此承诺完成时间,在原本闲置的时段取得的进展即使时机不确定,仍然具有价值。

AWS 演示带来的更广泛启示是:不同工作负载可以对利用率和可预测性采取不同处理方式。管理员不必在所有地方施加同一种取舍。

恢复质量决定共享能否带来有价值的进展

调度程序可以将被中断的作业返回队列,而不保证应用程序保留其先前的工作成果。

AWS 的作业抢占行为明确说明了执行语义。抢占会停止处于已调度、启动或运行状态的工作,以便为可运行的作业腾出空间。默认情况下,被抢占的作业会返回可运行状态,且没有抢占次数限制。

管理员可以设置 preemptionRetriesBeforeTermination 来限制这种行为。值为零会使作业在首次被抢占时失败。AWS 还表示,其选择启发式算法会考虑借用的实例资源、相对优先级和作业持续时间。

这些细节说明,“最高优先级优先运行”需要附加条件。资源需求和共享安排都会参与调度决策。优先级是一项输入,而不是任何紧急作业都能立即启动的普遍承诺。

应用程序层面的关注点在于恢复。作业可能具备再次运行的资格,却仍缺少高效继续执行所需的状态。排队与恢复解决的是相关但不同的问题。

AWS 的训练检查点支持会保存模型状态,并将配置的检查点文件与其对象存储服务 Amazon S3 同步。自定义训练代码必须正确保存和加载相关状态。AWS 还将检查点频率的控制权交给训练脚本。

这意味着,借用策略应建立在已验证的恢复路径之上。仅启用一个存储目标,并不能证明某个训练应用程序能够重建所需的一切。

对于假设性的实验,模型参数可能只是所需状态的一部分。应用程序所有者应检查其框架所要求的完整续跑行为。合适的测试是:中断执行后,是否能取得可接受的进展并产生有效结果。

检查点频率还会带来另一项权衡。更频繁地保存可以保留更近的进展,但保存本身也会消耗资源。合适的间隔取决于工作负载,而非某种通用的调度偏好。

这些是工程层面的考量,而不是 AWS 功能失效的证据。它们解释了为何利用率相关的主张需要工作负载测量。繁忙的实例并不自动意味着完成了更多训练。

有价值的评估应包括重复计算、重启准备、中断后的等待时间以及成功完成情况。仅关注被占用的资源,可能会忽略获得这种利用率的成本。

当团队将每个可运行的工作负载都视为同样适合借用时,风险最大。有些实验能够很好地容忍碎片化访问;另一些实验的恢复开销足够高,以至于等待稳定访问成为更实际的选择。

反复中断也值得单独关注。一个不断恢复的作业可能消耗资源,却仍错过其完成窗口。重试限制提供了一种控制手段,但选择该限制需要在达到上限时制定运营响应。

必须有人负责这一响应。失败应促成关于重新分类、重新提交或采用不同容量安排的明确决策。否则,有界重试策略只会将协调负担转移到事件处理中。

受保护策略同样存在附加条件。AWS 记录称,降低某个共享的容量限制,可能会使其现有工作具备被抢占的资格。因此,管理员应将分配变更视为会影响执行的变更。

团队对连续性的预期取决于其共享策略和分配的稳定性。配置审查很重要,因为一次策略编辑就可能改变运行中工作保持受保护的条件。

AWS 已发布的示例是一种运营演示,而不是对客户节省的独立基准测试。它提供了一种评估机制,但并未对吞吐量或完成时间给出普遍答案。

最有力的采用理由将来自那些在真实竞争条件下测量过恢复行为的工作负载。这样的证据将调度灵活性与有用结果联系起来,而不是假定两者必然同步变化。

Kueue 为共享训练基础设施提供了对照

AWS 在托管训练中封装了一个熟悉的资源共享问题,而 Kubernetes 用户则可通过 Kueue 选择另一条路径。

Kueue 是一个 Kubernetes 原生的工作负载排队系统,记录了相关队列之间的借用和抢占机制。其抢占策略区分了队列内部的中断与涉及其他队列的回收。它还提供了不同方法,以共享对可借用资源的访问权。

这一对照将 AWS 功能置于更广泛的背景中。对未使用分配的条件性访问是一种成熟的调度方法。AWS 的发展之处在于,将这种方法引入使用 SageMaker Training 的团队工作流中。

因此,相关比较应聚焦运营层面。已经采用托管训练的团队,需要评估 Batch 对其分配协议的表达效果。运行 Kubernetes 的团队,则要将这些协议与现有工作负载控制器和平台责任一并评估。

两种产品的文档都没有确立一个普遍的胜者。现有证据支持比较策略适配性和运营责任,而不是宣称某一系统始终能实现更高利用率。

这很重要,因为组织可能会过于狭隘地界定调度决策。借用很有吸引力,但调度程序只是训练环境的一部分。执行工具、恢复实践、访问控制和支持责任都会影响结果。

对于 SageMaker 用户,将资源分配整合到现有训练路径中,能够减少管理员需要协调的独立系统数量。这一收益是否足以证明采用该功能合理,取决于组织当前的工作流。

对于 Kubernetes 用户而言,仅为了配额共享而引入独立的托管训练路径,需要更广泛的评估。功能对比本身无法涵盖变更执行环境所带来的影响。

核心问题始终相同:当容量所有者处于空闲状态时,已分配的容量应保持有用,同时所有者保有可信的恢复路径。两种方法都要求团队决定临时访问允许什么。

这也说明为何严格隔离仍是合理的比较选项。当中断风险会带来更严重后果时,组织可以理性地接受闲置资源。利用率只是多个目标之一。

反过来,拥有灵活实验的组织也可以理性地优先考虑机会性进展。其工作负载和交付承诺支持不同的平衡方式。调度策略应表达这种平衡,而非强加一个抽象的利用率目标。

当 GPU 容量共享能以人们理解的策略取代现有协调负担时,它最具说服力。当组织无法说明哪些工作应让出资源或进行恢复时,它的说服力就会减弱。

因此,竞争压力部分落在内部平台实践上。维护手动分配流程的团队现在有了一个可评估的托管选项。他们需要证据证明,当前方法提供的收益值得其协调成本。

AWS 的功能并未消除这些比较的必要性。它为 SageMaker 用户提供了另一种实现配额权益与临时访问之间协议的方式。

关注完成时间、恢复行为和策略稳定性

最有价值的证据将把资源共享与已完成的工作及分配团队可预测的访问联系起来。

首要关注信号是在真实竞争条件下的完成时间。一项有意义的评估应比较工作何时完成,包括出借团队重新使用资源时会发生什么。仅凭更高的占用率只能提供不完整的证据。

如果借用能够加速灵活工作,同时为分配团队保留可接受的访问条件,采用理由就会增强。如果中断延迟抵消了收益,管理员就有理由缩小可借用的工作负载范围。

第二个信号是代表性应用程序的恢复行为。中断后成功续跑将支持更广泛地使用借用。反复重启、无效续跑或大量重复计算,则会削弱这一理由。

这些证据需要应用程序背景。一个训练脚本上的成功演示,不能证明每个模型工作流都会以相同方式运行。团队应记录与每种工作负载分类相关联的恢复假设。

第三个信号是用户开始依赖共享访问后策略的稳定性。管理员应关注分配规则是否仍然易于理解,还是频繁产生例外、优先级变更和争议。

稳定的策略表明,组织已将其优先事项转化为可用的调度规则。持续的覆盖操作则表明,配置的共享与实际承诺之间存在不匹配。

未来几个月,这些信号比启用该功能的团队数量更重要。采用表明兴趣;完成行为、恢复质量和可管理的治理,才能证明这种安排是否有效。

10 月的演示为平台团队提供了评估的具体起点。它按工作负载对中断的容忍度进行区分,并展示了回收如何改变执行。下一步取决于组织自身的工作负载证据。

对于工程领导者,决策应将基础设施行为与交付承诺联系起来。训练环境可以支持更多机会性工作,而不让每个完成期限都同样可预测。

对于研究人员,直接含义是,更早获得访问权可能伴随有条件的连续性。实验计划应反映已分配资源与借用资源之间的差异。

从团队了解其中断行为的代表性工作负载开始。记录其完成时间、恢复结果,以及分配团队收回容量所产生的影响。在扩大访问范围之前,将这些结果与现有安排进行比较。将分配协议与结果一同保留,以便审查者判断策略是否实现了预期目的。AWS Batch 配额管理提供了一种自动化出借和回收的具体方式,但应由你的工作负载证据决定它适用于何处。哪些实验可以接受临时访问,哪些实验需要更可预测的执行窗口?

 
 

免费开始使用

一款本地优先的AI助手

为了获得更好的人工智能体验,

remio 目前仅支持Windows 10+ (x64)和M-Chip Mac 。

你的 AI 工作伙伴

和 remio 一起高效工作

规划、创作、交付

一站式完成

bottom of page