最近在试着把Llama 3.1 8B部署到自己的单卡3090上做API服务。按照常规FP16加载,光是模型就占了16GB,再加上kv cache,跑个并发生成batch size=2就直接OOM了。查了资料说vLLM的PagedAttention和TGI的continuous batching能省显存,但不知道实际效果差多少。有没有大佬实测过这两种框架在单卡24G显存下的极限并发数?另外,量化到int4的话,会不会影响长文本生成质量?我主要做对话和摘要,求指点。
楼主
2026-07-28
部署Llama 3.1 8B时显存爆了,用vLLM还是用TGI更省显存?
请 登录 后发表回复
全部回复
共 142 条
2楼
2天前
int4跑对话摘要基本没差,长文本摘要才容易掉点,建议先试vLLM的AWQ。
3楼
1天前
我3090上跑过类似配置,FP16下8B模型加vLLM,batch size开到4左右就开始紧张了,PagedAttention确实能省一些碎片,但别指望翻天覆地的变化。TGI我也试过,continuous batching在并发上来之后吞吐更稳,但显存峰值反而比vLLM略高一点,可能跟它的调度策略有关。单卡24G想跑大batch,老实说瓶颈还是权重本身占的那16G,量化几乎是绕不开的路。int4我做过对话和摘要的对比,AWQ量化后长文本摘要偶尔会丢细节,尤其是超过4k token的输入,对话倒还好,基本感知不出来。如果你主要做API服务,建议vLLM加AWQ,并发和显存平衡得比较好,TGI更适合需要动态批处理且请求长度差异大的场景。另外3090不支持bf16原生加速,记得确认下框架有没有走fp16路径,这个也会影响实际显存占用。