🐸 青蛙编程网

推理服务 vLLM

vLLM = 高性能的大模型推理与服务引擎。它把模型权重装进显存,用上 GPU 的每一分算力,再开一个 HTTP 服务——接口和 OpenAI 一模一样,你的代码换个 base_url 就能连。

前置:大语言模型 LLM(知道 API 长什么样即可)、Linux 基础。 基线:vLLM 0.32.0(Python ≥3.11 且 <3.15)· PyTorch 2.14.1 · CUDA 12.x

为什么要自己部署

上一章讲过「学习阶段为什么用 API」。但一旦用量上来了,自建就有意义了:

场景 优势
数据不能出内网 模型和推理都在自己机器上
调用量巨大 自建的单位成本更低
需要专用模型 微调过的、或者自己蒸馏的模型,云上没有
延迟要自己掌控 不用排队、不受服务商限流

vLLM 的核心本事

能力 说明
PagedAttention 把 KV cache 分页管理,显存利用率大幅提升,显存不够的模型也能跑起来
连续批处理 有请求就调度、有请求完成就换下一个,GPU 不空转
OpenAI 兼容接口 vllm serve 起来就是标准 OpenAI API,官方 SDK 改个 base_url 就能用
量化支持 FP16 / FP8 / INT4,多种精度可选
多模型多卡 张量并行、流水线并行,横向扩

学习目标

  • 知道 vLLM 在整个推理链条上的位置
  • 会装 vLLM、用 vllm serve 起一个模型
  • 会用 OpenAI 兼容接口连上自己的服务
  • 会看显存占用、批处理参数,排查 OOM

技术路线

章节 核心内容
1. vLLM 是什么 定位、原理、与 Ollama 的差别
2. 环境搭建 装 CUDA、装 vLLM,跑起第一个模型
3. 启动服务 vllm serve 与 OpenAI 兼容接口
4. 性能调优 显存、批处理、量化、并行
5. 部署与运维 健康检查、日志、多卡与容器
6. 综合练习 把自有模型接进你的应用

硬件门槛提醒:vLLM 是给有显卡的机器准备的。没有 NVIDIA 显卡的话,本教程的命令跑不起来——但第 1、3 章的原理和接口部分仍然可读,看完你就知道自建和调 API 到底差在哪。

从 1. vLLM 是什么 开始吧!