OOM逼出的调度器:AI agent也能自己排队训练模型了

原创
alex 1天前 阅读数 8 #头条
你同时往一张GPU上塞几个训练任务,结果OOM把整批活干崩了——这种事谁没遇到过?一位开发者就是这么被逼出来的。他一边折腾几个简单模型,一边撞来撞去全是显存爆掉。于是他动手写了个简易的GPU任务调度器,还留了个让AI agent能自主调度的接口。Claude帮他写了大部分代码,但他自己把整个仓库跑通才算完。在我看来,这事有意思的地方不在于工具多复杂,而在于需求确实真实存在:单卡上缺的往往就是一个不会炸的训练队列,现有的调度系统要么太重,要么压根没为个人开发者设计。

原文:Show HN: A Simple GPU Job Scheduler · 来源:Hacker News

版权声明

所有资源都来源于爬虫采集,如有侵权请联系我们,我们将立即删除