2026年7月9日 周四晚上19:30,报名腾讯会议了解“如何构建自进化的动态知识库(Brain)”(限30人)
免费POC, 零成本试错
FDE知识库

FDE知识库

学习大模型的前沿技术与行业落地应用


收藏

为什么vLLM做不到?解密Ollama越级部署黑科技:以DeepSeek-R1-8B为例

发布日期:2025-03-08 11:11:56 浏览次数: 4198
作者:5ycode

微信搜一搜,关注“5ycode”

推荐语

探索vLLM与Ollama在大型模型部署上的技术差异,揭示背后的显存优化黑科技。

核心内容:
1. vLLM与Ollama在70B模型部署中的性能对比
2. DeepSeek-R1-8B显存需求分析与Ollama的显存优化技术
3. vLLM的显存管理策略及其对性能的影响

杨芳贤
53AI创始人/腾讯云(TVP)最具价值专家

 

最近在折腾双4090GPU,近200g内存服务器vllm部署70b的实验。使用ragflow知识库。

按照我之前使用ollama的理解,我觉的部署70b应该没啥问题,然后一个个的坑。

先说下

  • • 部署30b,还是很流畅的,但是推理能力确实比较弱
  • • 部署70b,通过不断地优化参数,最大tokens,4096 tokens没问题(还部署了一个bge-m3)
  • • 但是在ragflow中使用的时候,由于超出最大tokens,经常报oom,因此给ragflow的官方提交了一个issue。官方很快就修复了,这块还是很赞的。

越折腾疑问越多,然后我就了解下了一些知识点,有了这篇水文。

我们以RTX3060 12G显存为例拆解下。

DeepSeek-R1-8B的显存解剖

原始显存需求

  • • 权重(直接成本):8B参数 × 2字节(FP16)= 16 GB
  • • KV Cache(上下文缓存)
    • • 动态增长:2 × 层数 × 头数 × 头维度 × 序列长度 × 批次大小 × 精度
    • • (以2048 tokens为例): 2(K/V) × 32层 × 32头 × 128头维度 × 2048序列长度 × 2字节 = 1.07 GB
  • • 总需求:16 + 1.07 ≈ 17.07 GB(远超RTX 3060的12GB上限)

Ollama的“瘦身魔法

  • • 4-bit GPTQ量化:权重显存降至8B × 0.5字节 = 4 GB
  • • 动态卸载策略:将部分权重临时转存至CPU内存,显存占用降至6.2 GB(实测数据)
  • • 代价:CPU-GPU数据传输使Token生成速度从45 tokens/s降至28 tokens/s

最疯狂的是,我一个前同事用ollama,在macbook air 8g内存上部署了70b的模型。

vLLM的“显存洁癖"

  • • 设计原则:vllm追求极致吞吐量,拒绝任何可能影响性能的动态卸载
  • • 显存硬门槛:要求权重+KV Cache完全驻留GPU,导致DeepSeek-R1-8B在12GB显卡上无法启动

Ollama越级部署的三大核心技术

混合精度量化(灵活度碾压vLLM)

  • • 层级敏感量化:对底层MLP层使用4-bit,顶层Attention保留6-bit(减少精度损失)
  • • 实测对比(DeepSeek-R1-8B生成任务):
量化方案
显存占用
PPL(困惑度)
Ollama混合精度
6.2 GB
7.1
vLLM官方INT8量化
10.5 GB
6.9

内存-CPU分级存储(vLLM的禁区)

  • • 策略:将FP16的Attention权重保留在显存,MLP权重动态加载至内存
  • • 技术代价
    • • 每次前向传播增加5-8ms的PCIe传输延迟
    • • 但显存需求直降40%(从10.5GB→6.2GB)

自适应序列切片

  • • 长文本处理:当输入超过512 tokens时,自动拆分为多段并行处理
  • • 显存优化效果:2048 tokens输入时,峰值显存降低32%

vLLM为何“宁死不做越级部署”?

设计目标的根本冲突

  • • vLLM的核心使命:服务化场景下的高吞吐、低延迟(如百人同时访问的API)
  • • 拒绝动态卸载的原因
    • • CPU-GPU数据传输会严重拖慢并发请求的处理速度
    • • 显存碎片化可能破坏连续内存分配机制(vLLM依赖的PagedAttention技术)

量化支持的局限性

  • • 仅支持静态INT8:无法像Ollama混合使用4/6-bit,导致显存压缩率不足
  • • 校准数据固化:vLLM要求离线量化,而Ollama支持运行时动态调整

硬件兼容性差异

  • • Ollama的“妥协艺术”
    • • 为兼容消费级显卡(如RTX 3060),允许牺牲速度换取显存
    • • 甚至支持通过系统内存模拟显存(性能下降但能运行)
  • • vLLM的“精英主义”
    • • 仅优化Tesla系列显卡(如A100/H100),依赖高带宽显存
    • • 在消费卡上性能反而不如Ollama(RTX 4090实测低15%吞吐量)

实战测试——RTX 3060上的生死对决

测试环境

  • • 显卡:NVIDIA RTX 3060 12GB
  • • 测试任务:DeepSeek-R1-8B生成512 tokens回答

结果对比

框架显存占用生成速度可用性
vLLM
报错退出
-
完全不可用
Ollama
6.2/12 GB
22 tokens/s
流畅运行
原版Hugging Face
17.1/12 GB
报错退出
不可用

关键结论

  • • Ollama的生存逻辑:通过量化+动态卸载,将显存需求压缩至硬件的60%以下
  • • vLLM的哲学缺陷:为追求工业级性能,放弃对资源受限场景的适配

开发者选型指南

选Ollama的场景

  • • 个人开发者/小团队:硬件有限(≤24GB显存)
  • • 需要快速验证模型效果,对延迟容忍度高
  • • 长文本生成需求(利用切片策略降低峰值显存)

选vLLM的场景

  • • 企业级API服务:需要支持高并发(≥100 QPS)
  • • 拥有A100/H800等专业显卡,追求极致吞吐量
  • • 需兼容现有Kubernetes集群调度系统

终极避坑建议

  • • 警惕“虚假越级”:部分工具声称支持低显存运行,实则大幅裁剪模型参数(如DeepSeek-8B被阉割成6B)
  • • 验证量化完整性:使用llm-int8工具检查Attention层是否真的保留高精度
  • • 压测保平安:对Ollama需测试长时生成的显存泄漏问题(部分版本存在累积占用bug)

结语:没有神话,只有取舍

Ollama的“越级”本质是技术民主化——让更多人用上大模型,哪怕牺牲速度;vLLM的“高冷”则是商业现实的抉择。未来二者的融合或许会出现(如vLLM引入动态卸载),但在此之前,开发者仍需认清需求,选择最适合的战场。

相关术语

内存(RAM)与显存(VRAM)

  • • 系统内存(RAM):由CPU直接管理的主内存,通常称为“内存”,物理上通过主板插槽连接,供所有系统进程共享。
  • • 显存(VRAM):GPU专用内存,通过PCIe总线与CPU通信,专为高吞吐并行计算设计。
  • • 关键区别
    • • CPU不直接拥有内存,而是通过内存控制器访问RAM;
    • • GPU显存是独立硬件,与CPU内存物理分离。

Ollama显存优化的本质:CPU-GPU异构内存交换

当Ollama声称“将部分权重转存至CPU内存”时,其技术本质是:
将GPU显存中暂时不用的权重数据,通过PCIe总线转移到系统内存(RAM),并在需要时动态加载回显存
这一过程涉及以下核心技术:

(1)内存分级策略(Memory Tiering)

  • • 热数据:当前计算所需的权重(如正在执行的Attention层)保留在显存。
  • • 冷数据:后续步骤才需要的权重(如下一层的MLP参数)暂存至系统内存。
  • • 交换粒度:通常以层(Layer)为单位,例如将DeepSeek-R1-8B的32层分成多个组按需加载。

(2)预取与缓存(Prefetching & Caching)

  • • 预加载机制:在计算当前层时,异步将下一层权重从系统内存提前传输至显存。
  • • 缓存策略:对频繁使用的权重(如Embedding层)在显存中永久保留副本。

(3)硬件加速传输

  • • Pinned Memory:使用torch.cuda.Stream配合锁页内存(Pinned Memory),减少CPU-GPU数据传输延迟。
  • • Direct Memory Access(DMA):绕过CPU直接由GPU控制器管理传输,实测带宽可达PCIe 4.0 x16的32GB/s。 

53AI,企业落地大模型首选服务商

产品:场景落地咨询+大模型应用平台+行业解决方案

承诺:免费POC验证,效果达标后再合作。零风险落地应用大模型,已交付160+中大型企业

联系我们

售前咨询
186 6662 7370
预约演示
185 8882 0121

微信扫码

添加专属顾问

回到顶部

加载中...

扫码咨询

扫码登录
登录即表示您同意《53AI网站服务协议》
服务协议

欢迎您使用【53AI 官方网站】(以下简称“本网站”或“我们”)。本《会员服务协议》(以下简称“本协议”)是您(以下简称“会员”或“用户”)与【深圳市博思协创网络科技有限公司】之间关于注册、登录及使用本网站会员服务所订立的法律协议。

在您注册或登录前,请务必审慎阅读、充分理解各条款内容,特别是免除或限制责任的条款、知识产权条款、争议解决条款等。此类条款将以加粗形式提示您注意。 当您通过微信公众号授权、手机验证码验证或其他方式成功登录本网站时,即视为您已完全理解并同意接受本协议的全部内容。

一、 定义

本网站:指由【深圳市博思协创网络科技有限公司】运营的,域名为【53ai.com】的网站及相关移动端页面。

会员服务:指本网站向注册会员提供的知识库文章查阅、内容检索及其他相关增值服务。

知识库内容:指本网站发布的包括但不限于文字、图表、数据、研究报告、行业分析等数字化内容资源。

二、 账号注册与登录

登录方式:本网站支持以下登录方式,您可根据实际情况选择:

微信公众号授权登录:您同意将您的微信OpenID信息授权给本网站,用于创建或关联会员账号。

手机验证码登录:您需提供真实有效的手机号码,并通过短信验证码完成身份验证与登录/注册。

账号安全:您的账号仅限您本人使用,禁止赠与、借用、租用、转让或售卖。因您保管不善导致的账号被盗、密码泄露等损失,由您自行承担。

实名认证:根据相关法律法规要求,我们可能要求您在特定功能下完成实名认证。如您拒绝提供,可能无法使用部分或全部服务。

未成年人保护:若您未满18周岁,请在法定监护人的陪同下阅读本协议,并在征得监护人同意后使用本服务。

三、 服务内容与规范

知识库查阅权限:会员登录后,有权按照其会员等级对应的权限范围,在线浏览、检索本网站知识库中的相关文章及内容。

服务变更:我们有权根据业务发展需要,调整、变更或终止部分服务内容,并将以网站公告、公众号消息等方式提前通知。

禁止行为:您在使用服务时不得实施以下行为:

利用技术手段批量爬取、下载、转存知识库内容;

将知识库内容用于商业目的或未经授权地向第三方传播;

干扰本网站正常运行或侵犯其他用户合法权益;

发布违法违规信息或从事违反公序良俗的活动。

四、 知识产权声明

权利归属:本网站知识库中的排版设计、软件代码等内容的知识产权均归【公司全称】或原权利人所有,受《中华人民共和国著作权法》等法律保护。

有限许可:本网站授予会员一项非独占、不可转让、不可转授权的普通许可,仅限于个人学习、研究之目的在线查阅知识库内容。

侵权追责:未经书面许可,任何单位或个人不得以任何形式复制、转载、摘编、镜像、汇编或以其他方式使用上述内容。一经发现,我们保留追究其法律责任的权利。

五、 个人信息保护

我们重视对您个人信息的保护。关于我们如何收集、使用、存储和保护您的个人信息,请单独阅读 《隐私政策》。

您通过微信公众号授权或手机号验证所提供的信息,我们将严格按照《个人信息保护法》的规定处理,仅用于身份识别、服务提供及安全验证等必要用途。

您可以随时通过网站设置或联系客服行使查阅、更正、删除个人信息及撤回授权同意的权利。

六、 免责声明

内容准确性:知识库内容仅供参考,不构成专业建议。我们不对其完整性、准确性、时效性作任何明示或暗示的保证,您应自行判断并承担使用风险。

不可抗力:因自然灾害、政策法规变化、网络故障、第三方平台接口异常(如微信接口维护、运营商短信通道故障)等不可抗力导致的服务中断或延迟,我们不承担违约责任。

第三方链接:本网站可能包含指向第三方网站的链接,该等网站的内容和服务不受我们控制,请您自行甄别风险。

七、 违约责任

如您违反本协议约定,我们有权视情节采取警告、限制功能、暂停服务、注销账号等措施,并保留要求赔偿损失的权利。

如因您的违约行为导致我们遭受行政处罚、第三方索赔或商誉损失,您应承担全部赔偿责任(包括但不限于罚款、赔偿金、律师费、公证费等)。

八、 法律适用与争议解决

本协议的订立、执行和解释均适用中华人民共和国大陆地区法律。

因本协议产生的或与本协议有关的任何争议,双方应友好协商解决;协商不成的,任何一方均可向【公司所在地】有管辖权的人民法院提起诉讼。

九、 其他

本协议构成双方就本服务达成的完整协议,取代此前任何口头或书面约定。

本协议任一条款被认定为无效或不可执行的,不影响其他条款的效力。

我们对本协议享有最终解释权,并在法律允许的范围内保留随时修改的权利。修改后的协议一经公布即生效,继续使用服务即视为同意修订内容。


已查阅