显存与并发
BF16、FP8 和 4 位权重按每参数 2、1.08 和 0.56 字节估计。GQA 和 MQA 的单卡每 Token KV 字节数为层数 × 2 × 单卡 KV 头数 × 头维度 × 元素字节数。张量并行优先切分 KV 头,不能继续切分时则复制。MLA 的压缩缓存会在每张张量并行卡上保留一份。估算容量时,将输入与生成长度之和向上对齐到 16 Token 的缓存块。
大模型推理部署
输入 GPU 配置、模型和请求特征,估算显存、并发请求数、每秒 Token 数和批量任务耗时。
1 张 GPU · 200 生成 Token / 请求
容量取决于 KV 缓存和引擎并发上限。当前并发还受任务量和目标并发限制。
BF16、FP8 和 4 位权重按每参数 2、1.08 和 0.56 字节估计。GQA 和 MQA 的单卡每 Token KV 字节数为层数 × 2 × 单卡 KV 头数 × 头维度 × 元素字节数。张量并行优先切分 KV 头,不能继续切分时则复制。MLA 的压缩缓存会在每张张量并行卡上保留一份。估算容量时,将输入与生成长度之和向上对齐到 16 Token 的缓存块。
每阶段取算力耗时与显存读写耗时的较大值。输入处理按批处理并产生首 Token。解码时,同批请求共享权重读取,KV 按平均上下文长度读取。MoE 模型按激活参数量估算计算开销,按独立均匀的专家路由估算权重读取量。
含输入处理的吞吐用每批生成 Token(回答加思考)除以整批耗时。任务耗时累加完整批次和尾批。请求均衡分配给各副本。
适用于同型号 GPU 上、权重常驻显存的文本生成。估算省略排队、CPU 卸载、专家并行、前缀复用和连续批处理。推理内核、量化版本和互联会影响结果。若设备报告的显存与标称容量不同,可改用自定义 GPU。请使用代表性请求做 vLLM 服务压测。若 Qwen 部署开启了 YaRN,可用自定义模型填写实际上下文上限。