GPT中文小秘书挑几下就能开工
我积分

算大模型显存和并发预算

关于「算大模型显存和并发预算」:能拿到什么、要点选什么、怎么办

卡型参数量一填,显存并发算明白。显存预算估算是按卡型、参数量与精度算权重和KV缓存;在 GPT中文网点选几项,能直接拿到显存与并发估算表、vLLM 配置草稿。

能拿到什么

  • 推荐:显存与并发估算表 + vLLM 配置草稿(表内分行列权重、KV、余量;草稿含启动参数与并发建议)

要你点选的几项

卡型、卡数量、模型参数量、精度/量化、最大序列长、目标并发数、KV缓存精度、模型结构。多数点一下就行,没选的按常见情况替你定,交付时说清哪几项是按默认的。

怎么办

  1. 先按参数量乘每参字节数估权重显存
  2. 按层数头数头维序列长批大小算 KV 缓存
  3. 总量扣减后算剩余显存占比
  4. 按剩余显存反推最大并发并留余量
  5. 生成 vLLM 启动参数草稿

别人还这样说

  • 这个参数量配 INT8 大概能开多少并发
  • 给我一版 vLLM 的启动参数草稿,要留余量
直接说你要办的事。下面是别人怎么办的,点任何一项就换成你自己的。左边是别人怎么办的,右边是你的,点一项就开始改。
示范 · 别人这样说

「帮我按 80G 卡和 70B 模型算下显存够不够跑」

替他选了:70B

按示范选好的 点一项就变成你的

你的成品会在这里出现
示例成品 · 不是你的按示范写的,看个样子
示例:显存与并发估算表(示意) 卡型 | 单卡显存 | 卡数 | 权重显存 | KV缓存/请求 | 可开并发 | 剩余占比 A100 80G | 80GB | 1 | 约28GB(70B INT4) | 约1.2GB(4K序列) | 约30 | 约20% 说明:权重显存≈参数量×每参字节数;KV缓存≈层数×2×头数×头维×序列长×批大小×字节数;剩余显存占比按总量扣减。以上数字为示意,实际以程序计算为准。 vLLM 配置草稿(示意) 模型路径:/models/xxx 张量并行:1 dtype:half 最大模型长度:4096 显存利用率:0.90 gpu_memory_utilization:0.90 max_num_seqs:24 量化:awq 结论由持证人员核对后才能用。

GPT中文 也可能出错,成品请自己核对关键数字和事实。条款 · 隐私