一、模型名称拆解(选型第一步)

格式系列-版本-参数量-后缀

部分示例含义
系列Qwen, Gemma, Llama研发团队
版本3, 2.5, 4代际,数字越大越新
参数量0.6B, 8B, 26BB=10亿参数,决定智商与资源
后缀见下表优化方式

二、精度与显存换算(理解后缀的前提)

模型权重显存 = 参数量(B) × 每参数字节数

精度每参数字节示例:8B模型权重
FP16 / BF16216 GB
FP818 GB
INT4 / NVFP40.54 GB

量化后缀(如AWQ、NVFP4、FP8)的本质就是降低每参数的字节数,从而减少显存占用。


三、常见后缀速查(2026·详细版)

后缀本质精度/格式显存占比(相对FP16)适用场景注意事项
AWQ / GPTQINT4整数量化4bit整数25%(4GB/8B)通用首选,兼容性最好vLLM/Ollama均完美支持
NVFP4NVIDIA FP4浮点量化4bit浮点25%(4GB/8B)RTX 50系专属,速度最快需TensorRT-LLM或vLLM≥0.9
FP8FP8浮点量化8bit浮点50%(8GB/8B)长上下文首选,精度损失极小40系/50系均支持,KV Cache压缩标配
GGUFCPU/GPU混合推理可调(常用Q4_K_M等)灵活(可设层数)显存<8GB 或 Mac 用户必选用Ollama/llama.cpp,非vLLM
A4B / MoE混合专家架构总参数不变总参数×精度(26B约13GB FP16)推理快(激活少),显存不省26B-A4B仍需加载26B权重,勿被“A4B”误导
Distill知识蒸馏小模型独立训练的小模型原生小模型大小(如1.7B约3.4GB FP16)低配设备的高质量选择不是压缩,是“徒弟学师傅”

⚠️ 关键辨析:量化 vs 蒸馏

维度量化 (Quantization)蒸馏 (Distillation)
通俗类比高清照片压成JPG,文件变小,画质微损老师傅带徒弟,徒弟用更少脑细胞学会本事
模型参数数量不变(8B量化后还是8B个参数)大幅减少(8B → 1.7B)
是否需要训练❌ 不需要,下载即用✅ 需要大量数据重新训练
智商影响4bit几乎无损(差距<1%);极端压缩才变笨取决于学生容量,可能接近甚至超越同尺寸原生模型
解决什么问题显存不够装原版,想省显存、提速设备太弱跑不动大模型,想造出更强的小模型

实战组合

  • 显存不够跑某个特定模型? → 找它的量化版(Qwen3-8B-AWQ)
  • 设备太弱,连量化版都跑不动? → 找蒸馏小模型(Qwen3-1.7B-Distill)
  • 既要小又要强? → 蒸馏模型的量化版(Qwen3-1.7B-Distill-AWQ)

四、显存估算(模型权重 + KV Cache)

总显存 ≈ 模型权重 + KV Cache + 预留1~2GB

模型权重

直接由参数量和精度决定(见第二部分)。

KV Cache(键值缓存)

实际作用:大模型生成文字是一个字一个字往外蹦的。如果没有KV Cache,每生成一个新字,模型都要把前面所有的对话重新计算一遍——相当于写作文时每写一个字都要从头重读整篇文章,效率极低。KV Cache 会把之前已处理过的内容(Key和Value)缓存起来,每次只计算当前新字与前面内容的关联,用显存空间换生成速度

关键参数

  • --max-model-len:设上限(超出截断)
  • --quantization-kv-cache fp8:显存减半
  • --enable-prefix-caching:自动复用,省30-60%

实测占用(FP8,每万token)

模型类型注意力机制占用
Qwen3-8B / Llama-3-8BGQA~0.25 GB
Gemma-4-26B-A4BMQA~0.18 GB
老款模型MHA0.50.7 GB

建议:不要手算,用 vllm estimate --model <path> --max-model-len 32768 或在线计算器。


五、推理引擎选择(决策树)

你的显卡?
├─ RTX 50系 → TensorRT-LLM 或 vLLM≥0.9(NVFP4)
├─ RTX 30/40系 → vLLM(FP8/AWQ)
├─ 显存≤8GB / Mac / CPU → Ollama + GGUF
└─ AMD → vLLM (ROCm) 或 LM Studio

引擎定位

  • TensorRT-LLM:NVIDIA 官方优化引擎,RTX 50系 NVFP4 性能最佳。
  • vLLM:服务端引擎,适合 API 调用和高并发。
  • Ollama:桌面工具,适合个人对话和快速体验。

⚙️ TensorRT-LLM 的特殊说明(量化模型 & 蒸馏模型 & Gemma-4)

  • 对量化模型:TensorRT-LLM 原生支持 AWQ、FP8、NVFP4 等量化格式,但需要使用 NVIDIA ModelOpt 工具生成其专用的引擎格式,社区直接下载的量化模型(如 HuggingFace 上的 AWQ 模型)可能无法直接加载。
  • 对蒸馏模型:TensorRT-LLM 可以部署蒸馏后的小模型(如 Qwen3-1.7B-Distill),但蒸馏过程本身由 ModelOpt 或其他工具完成,TensorRT-LLM 只负责推理加速。
  • 关于 nvidia/Gemma-4-26B-A4B-NVFP4:目前不能直接使用 TensorRT-LLM 加载(因其后端兼容性问题),但 TensorRT-LLM 本身支持 Gemma-4 架构和 NVFP4 量化,正确做法是下载基础模型 google/gemma-4-26B-A4B-it,在 TensorRT-LLM 环境下使用 ModelOpt 自行量化为 NVFP4 格式并编译。简而言之:TensorRT-LLM 不是不能跑,而是需要“亲手转换”

六、vLLM 部署:模型大小与并发数的关系(新增核心章节)

6.1 核心逻辑:显存是连接模型与并发的桥梁

vLLM 将显存划分为:

  1. 模型权重(固定):模型参数占用的静态显存。
  2. KV Cache(动态):每个并发请求都会消耗 KV Cache 来存储上下文。
  3. 其他开销:激活值、框架开销等。

可用 KV Cache 显存 = (总显存 × gpu_memory_utilization) – 模型权重 – 其他开销

模型越大,权重占用越多,留给 KV Cache 的显存就越少,从而限制并发能力。

6.2 链条影响:模型大小 → 并发上限

  • 大模型(如 70B)权重可能需要 140GB 显存(FP16),即便量化后也占用巨大。
  • KV Cache 总量决定了可同时处理的 Token 总数,进而决定了最大并发序列数(max_num_seqs)。
  • 每个并发请求消耗的 KV Cache 由 max_model_len(上下文长度)决定,长度越长,单个请求占用越多,并发数就越低。

6.3 关键参数与调控

参数作用对并发的直接影响
max_num_seqs最大并发序列数直接上限,每增加1,KV Cache 线性增加
max_model_len最大上下文长度越大,每个请求占用 KV Cache 越多,减少可并发数
gpu_memory_utilizationvLLM 可用显存比例调高增加可用显存,但需预留系统缓冲
tensor_parallel_size张量并行度(多卡)增加总显存,为 KV Cache 腾出空间
quantization (如 AWQ/FP8)模型量化减少权重显存,间接增加 KV Cache 可用空间
--kv-cache-dtype fp8KV Cache 压缩直接减半 KV Cache 占用,提升并发能力

6.4 调优建议与估算工具

  • 启动时观察日志:vLLM 会打印 # GPU blocks: XXXGPU KV cache size,可评估当前配置下的并发潜力。
  • 使用官方 Colab 计算器:输入模型、上下文长度、批处理大小,估算所需显存和建议的 max_num_seqs
  • 显存不足抢救顺序(与之前章节一致):
    1. 开启 --enable-prefix-caching(零成本)
    2. KV Cache FP16 → FP8 → NVFP4(50系)
    3. 换 AWQ/GPTQ 量化模型
    4. 降低 --max-model-len
    5. 换 GGUF + Ollama 开启 CPU offload(牺牲速度)

七、实战落地:选型与排错(综合推荐)

7.1 根据显卡推荐

硬件推荐模型引擎体验
≤8GB / MacQwen3-0.6B-Distill 或 Qwen2.5-3B-GGUFOllama日常流畅
RTX 30/40 12-24GBQwen3-8B-AWQvLLM速度智商平衡
RTX 50 24GB+Gemma-4-26B-A4B-NVFP4(需自行转换)vLLM≥0.9 或 TensorRT-LLM极致速度

7.2 显存不足抢救顺序(通用)

  1. 开启 --enable-prefix-caching(零成本)
  2. KV Cache FP16 → FP8 → NVFP4(50系)
  3. 换 AWQ/GPTQ 量化模型
  4. 降低 --max-model-len(每减半省约一半KV Cache)
  5. 换 GGUF + Ollama 开启 CPU offload(牺牲速度)

7.3 并发能力不足(vLLM)抢救顺序

  1. 降低 max_num_seqs(直接减少并发数)
  2. 降低 max_model_len(减少每个请求的上下文长度)
  3. 启用 KV Cache 量化(--kv-cache-dtype fp8
  4. 考虑多卡并行(增大 tensor_parallel_size
  5. 换更小的模型或量化版本

📌 总结

这份笔记覆盖了从模型名称解读量化与蒸馏辨析显存估算KV Cache 原理推理引擎选择(含 TensorRT-LLM 的特殊性),最后新增了 vLLM 并发与模型大小的内在关系,形成了一个完整的闭环。无论你是个人开发者还是团队部署,都可以按图索骥,快速定位问题并优化配置。

如有未尽之处,欢迎随时补充完善!

更多推荐