我给 Colab 装了个 “智能电闸”
订了 Google One AI Premium (Google AI Pro) 送每月 200 Colab CU,为避免忘关机烧额度,我做了一个轻量网关,像“智能电闸”一样按需通电、空闲自动拉闸,还打通了 AI Agent 一键调度
音频
我给 Colab 装了个 “智能电闸”
0:00/0:00
✦ ✦ ✦
订了 Google One AI Premium (Google AI Pro) 的人,账户里通常都有个被遗忘的福利:
每月 200 个 Google Colab 计算单元 (CU)
换算一下,每个月能白嫖英伟达 24GB 显存的 NVIDIA L4 GPU 连续出图 100 多个小时
要是你手头是更顶格的 Google AI Ultra 套餐,每个月更是直接附赠高达 1,000 CU,连单卡 40G / 80G 显存的 A100 都能随便挥霍,FLUX 或者视频大模型放开手脚跑
但我敢说,90% 的人这笔算力都是月底默默清零
不是不想用,是原生的 Colab 跑生图太折磨了
原生 Colab 的三宗罪
试过的人都懂:
- 伺候 Colab 太麻烦——只是想画张图,非得跑到网页端去折腾 Jupyter 笔记本,还要学怎么挨个点单元格运行排错
- 忘关机被“偷家”——画完图随手把网页关了,机器还在后台跑,一觉醒来半个月额度扣光
- 不敢借人——想丢给脚本跑,或者给朋友用,得把自己的 Google 账号密码交出去
既然痛点这么明确,那就给它做个管家
我花时间写了个轻量调度网关:Comfy Colab Kit
其实就是给 Colab 装了个 “智能电闸”
亮点一:像电闸一样,用时通电,不用拉闸
原生 Colab 最大的坑,就是用和算账脱节
现在逻辑变了:
- 平时断电:GPU 处于深度休眠,0 算力开销
- 用时合闸:点一下唤醒,或者外部发了生图任务,自动去 Google 申请 L4 显卡
- 空闲拉闸:连续 30 分钟没有新任务,网关自动释放机器彻底关机
而且不用担心操作习惯被绑架:
没任务时它是个零开销的休眠卡片;一旦就绪,直通的就是原汁原味的官方 ComfyUI,既能用新手表单,也能随时切回节点画布
画完图拔腿就走,再也不用提心吊胆担心烧额度

冷启动唤醒网关
实时看环境安装和模型下载,今日消耗与剩余 CU 一目了然

唤醒后直通官方 ComfyUI
官方 ComfyUI 原生直通,新手用极简表单,老手随时切节点画布
亮点二:与 AI Agent 进行协作
机器跑在云端只是第一步
最有意思的,是把它接进我日常干活的 AI Agent 里面
丢进
~/.agents/skills/ 目录,或者网页端点一下“复制接入指令”,Agent 就能无缝认领它不仅知道怎么调用工具,还自动学会了整套生图 SOP:什么时候该查 CU 余额、文生图和图生图如何传参、什么时候唤醒和释放
以前生图要切窗口、调参数、存图片、再拖回项目
现在我的工作流变成了:动动嘴,AI Agent 自己去调显卡画图
全程不碰浏览器,也不手动执行任何 notebook 单元格
我负责构思和写代码,AI Agent 负责调动后台的 L4 显卡出图

文生图 · /comfy-colab-kit
写前端时跟 Agent 说一句:“帮我生成个科技感 404 插画”,41 秒完成出图并存入工程

图生图 · /comfy-colab-kit
随手画个草图丢给 Agent:“按原图构图换一套衣服”,自动上传原图完成精修渲染
亮点三:Profile 预设即开即用
玩过 ComfyUI 的人都知道,最折磨的往往不是生图,而是找模型和配节点
去 HuggingFace 翻几十 GB 的权重、挑 LoRA、对 VAE、再到处找别人分享的工作流 JSON 导入……哪怕少装一个自定义节点,满屏都是刺眼的红框报错
在网关里,我把这套繁琐操作全做成了 Profile 预设:
- 开箱即用热门工作流:内置了去条纹的 MOSS-Image、SD 3.5 Large、Qwen-Image 等精选配置,开机前点选即可
- 云端全自动就绪:依赖、权重与工作流在拉起 VM 时自动注入,跑完冒烟测试直接交到你手上
- 想加新模型极简:只需按规范写个声明式的 yaml 配置,后续下载与环境全交由网关自动托管
彻底告别了“每次开机先当半小时配电工”的折磨
当然你也可以直接 fork 这个项目,魔改属于你自己的工作流,甚至直接丢给 AI 帮你二次改造,都非常方便
亮点四:权限限制与零信任防护
自己用爽了,往往也想把链接丢给朋友一起蹭
但毕竟绑着自己的 Google 账号,权限必须卡得很死:
- 立在 Cloudflare Access 背后:公网访问走邮箱白名单验证码拦截,防网络扫描也防盗刷
- 分级权限控制:只有在 Owner 白名单里的邮箱才有权唤醒、切换模型和手动关机;其他人只能进来看图和提交任务,避免别人随意折腾你的算力
- 独立设备 Token 随时撤销:外部的 AI Agent 走专属设备码授权,需要时随时在网页端一键吊销
- Google 凭据物理锁死:OAuth 凭据只保存在受保护的数据卷内,无论是朋友还是外部 Agent,都绝对接触不到你的底层账号密钥
结语
不管是每月 200 CU 的 Pro,还是直接送 1,000 CU 的 Ultra
换算下来上百甚至几百个小时的纯 GPU 运算,日常用来给博客配图、UI 做原型,根本用不完
给 Colab 装上这个开关之后,这笔原本容易烂在账上的福利,终于成了我手头随时待命的私有算力池
项目完整代码和配置已开源,也搭了官网展示站:
- 💻 开源仓库:FireTable/comfy-colab-kit
手头有闲置 Colab 算力的朋友,不妨试试看
✦ ✦ ✦