最近在折腾基于Llama 3.1本地部署一个简单的AI Agent,用来做信息检索+总结。单用户测试还好,但一上到多轮对话(上下文积累到4k tokens以上)再加两三个并发请求,显存直接爆掉OOM。我试过vLLM和TGI,但感觉配置起来参数好复杂,什么max_num_batched_tokens、调度策略,看得有点懵。想问下有没有经验的同学,在不换显卡(目前是RTX 4090 24G)的前提下,怎么优化推理时的显存分配?或者有没有轻量的Agent框架能自动管理上下文长度和并发?先谢过各位前辈了。
楼主
2026-07-17
部署开源大模型做Agent,遇到并发推理+长上下文时OOM怎么破?
请 登录 后发表回复
全部回复
共 181 条
2楼
2天前
vLLM里把max_num_batched_tokens调小点确实有用,我4090上一般设2048到4096之间,再配合gpu_memory_utilization留点余量别拉满。长上下文那块可以考虑开enforce_eager省显存,虽然吞吐掉一些但至少不炸。Agent框架的话可以看看LangGraph,它本身不管显存但上下文裁剪逻辑好写,配合滑动窗口或者摘要压缩能省不少。你现在的OOM是发生在prefill阶段还是decode阶段,这个定位清楚了好对症下药。