最近在尝试把Llama 3 7B部署到我自己的机器上(RTX 3060 12G),用vLLM框架,Q4量化后能跑起来,但上下文一超过4K tokens就开始OOM。我主要想跑一些长文档总结,比如10K tokens左右。查了资料说可以用Flash Attention或者KV Cache复用,但配置起来有点懵。有没有老哥分享下实际部署中怎么压显存?或者换什么量化方案(比如AWQ、GPTQ)能多撑点长度?我现在连StreamingLLM都试了但效果有点玄学,求真实踩坑经验。
楼主
2026-07-30
部署7B大模型到本地,显存不够但还想跑长文本,有什么优化技巧吗?
请 登录 后发表回复
全部回复
共 142 条
2楼
26天前
AWQ比GPTQ省显存,我3060跑10K没问题,Flash Attention记得开,不然KV cache直接爆。
3楼
23天前
RTX 3060 12G跑7B的Q4确实挺紧的,你说的4K就OOM我一点都不意外,因为vLLM默认的gpu_memory_utilization是0.9,留给KV Cache的其实没多少。我自己试过最有效的组合是AWQ量化加--enable-prefix-caching和--max-model-len压到8192,然后gpu_memory_utilization调到0.95左右,能勉强撑到8K多,10K还是悬。GPTQ的话ExLlamaV2后端在长上下文上比vLLM省显存一些,但吞吐会掉,看你能不能接受。Flash Attention本身不省KV Cache的显存,它省的是中间激活那块,对长文本瓶颈帮助有限,别指望它解决OOM。StreamingLLM那套attention sink确实玄学,短文档总结容易丢关键信息,我后来放弃了。真想在12G上跑10K,建议考虑换Qwen2.5 7B或者干脆降到3B级别,再不行就上CPU offload,慢是慢但至少不炸。