v

vLLM

UC Berkeley 开源的高吞吐量 LLM 推理引擎,PagedAttention 技术实现 24 倍于传统推理的吞吐量,是生产环境部署大模型的首选方案

免费 🇨🇳 中文完全 ✓ 国内直连 API
vLLM 界面截图

工具简介

vLLM 是由 UC Berkeley 团队开发的开源大语言模型推理引擎,专为生产环境的高吞吐量、低延迟需求而设计。其核心创新——PagedAttention 注意力算法,通过将 KV 缓存(注意力机制中的键值对缓存)分页管理,大幅减少了显存碎片和浪费,在保持输出质量不变的前提下,将推理吞吐量提升至传统框架(如 Hugging Face Transformers)的 24 倍。

vLLM 已成为业界最广泛采用的 LLM 推理框架之一,被 Anyscale、AWS、Google Cloud、Modal、BentoML 等平台集成。GitHub 上拥有超过 40,000 颗星标,是 LLM 推理领域事实上的标准方案。

核心功能

  • PagedAttention 高效推理:通过操作系统式分页管理 KV 缓存,显存利用率提升至接近 100%,推理吞吐量可达传统方案的 24 倍。支持 continuous batching(连续批处理),保证请求不排队等待。
  • 广泛的模型支持:兼容 Hugging Face 上绝大多数 LLM 架构,包括 Llama、Mistral、Qwen、DeepSeek、Gemma、Phi、Falcon 等。支持 AWQ、GPTQ、FP8 等多种量化格式,可在一张消费级 GPU 上运行 70B 级模型。
  • OpenAI 兼容 API:内置 OpenAI-compatible API 服务器,无需修改任何代码即可用 OpenAI SDK 连接自部署的模型。支持流式输出、工具调用、多模态输入等完整功能。
  • 多 GPU 分布式推理:原生支持张量并行(tensor parallelism)和流水线并行(pipeline parallelism),可跨多 GPU 甚至多节点部署超大模型(如 405B Llama)。
  • 前缀缓存与推测解码:自动缓存重复前缀(系统提示词),减少冗余计算。支持推测解码(speculative decoding)加速自回归生成。

访问说明

vLLM 完全开源(Apache 2.0),通过 pip install vllm 即可安装。需要 NVIDIA GPU + CUDA 环境。官方提供 Docker 镜像简化部署。文档完善(docs.vllm.ai),社区活跃。

适合人群

  • AI 基础设施工程师:需要在生产环境部署高性能 LLM 推理服务
  • 后端开发者:需要自建 OpenAI 兼容的模型 API 端点
  • 研究团队:需要高效评估大量模型或处理批量推理任务
  • 成本敏感型团队:通过高效推理降低 GPU 使用成本

Related

相关工具

NEW

开源的 LLM 微调加速框架,将训练速度提升 2-5 倍、显存降低 50-80%,支持 Llama/Mistral/Qwen/DeepSeek 等主流模型的一键微调

免费增值 🇨🇳 中文完全 🔒 需要代理

Articles

相关文章