最近在折腾一个RAG+工具调用的Agent项目,用的Qwen2.5-7B-Instruct,量化到4bit后单卡3090勉强能跑。但问题是加上embedding模型和向量库,显存直接爆了,只能不停清缓存。我看很多人说用vLLM部署能省显存,但我试了下,配合LangChain的Agent时总报兼容性问题,要么是tool calling格式不对,要么是流式输出卡住。想请教下各位,本地部署大模型做Agent到底怎么选推理框架?是继续用transformers硬扛,还是换更小的模型(比如3B)?另外,有没有办法让embedding模型和LLM共享显存而不互相干扰?求具体可行的方案,谢谢!
楼主
2026-08-12
部署本地大模型做Agent,显存总是不够用,大佬们有什么优化思路吗?
请 登录 后发表回复
全部回复
共 62 条
2楼
23天前
我之前也踩过这坑,后来换成Ollama跑Qwen2.5-7B再挂个LlamaIndex做工具调用,显存稳多了,vLLM对LangChain的tool calling确实有点水土不服。embedding模型建议单独用CPU跑,比如bge-small这种,反正它推理量不大,省下来的显存给LLM更划算。要是还紧张就上3B,Agent场景下配合好prompt其实够用,别硬扛7B。
3楼
22天前
我之前也踩过这个坑,vLLM配LangChain的tool calling确实容易出幺蛾子,后来换成SGLang顺手不少,它原生支持OpenAI接口,Agent那边基本不用改。显存这块最有效的还是把embedding模型单独丢到CPU或者另一张小卡上,bge-small这类模型CPU推理完全够用,没必要跟LLM抢显存。7B换3B要慎重,工具调用能力掉得挺明显,不如先把上下文长度压一压,或者用AWQ量化试试。