实验室刚批下来一台A100 80G的服务器,想部署一个Qwen2.5-14B做内部知识库问答,并发量不大,大概同时5-10个人用。之前只玩过Ollama,但感觉性能不太行,网上看vLLM和TGI讨论比较多,有点纠结选哪个。主要诉求是稳定、部署别太复杂,最好能支持流式输出。有没有实际踩过坑的老哥说说,这俩在生产环境哪个更省心?显存占用差距大吗?
楼主
26天前
想在自己实验室服务器上部署大模型,选vLLM还是TGI比较好?
请 登录 后发表回复
全部回复
共 3 条
2楼
24天前
A100 80G跑14B其实挺宽裕的,你这场并发量真不算大,两个框架都能扛住。我自己两个都用过,vLLM的PagedAttention对显存利用率确实更狠一点,同样模型能塞下更多KV cache,不过你5-10个人用基本感受不到差别。TGI的好处是HuggingFace那套生态接得顺,镜像拉下来配几个环境变量就能跑,流式输出默认就有,踩坑相对少。vLLM部署也不复杂,但版本迭代快,偶尔会遇到依赖冲突或者某个模型要等新版本才支持,得有点折腾心理准备。稳定这块我个人感觉TGI更省心,尤其你不打算天天升级的话,跑起来基本不用管。不过Qwen系列在vLLM上的社区支持现在也很活跃,出问题搜得到答案。硬要选的话,图省事就TGI,想榨性能或者以后要上更大模型、更高并发就vLLM,你这场景我偏向TGI。
3楼
23天前
5-10个人用14B直接vLLM吧,A100 80G显存够得很,流式输出也稳,部署比TGI省心多了。
4楼
21天前
这个并发量其实两个都扛得住,A100 80G跑14B绰绰有余。我自己用vLLM多一些,主要是PagedAttention对显存利用确实友好,流式输出也稳。TGI的Docker部署更省事,文档清晰,但配置起来感觉没vLLM灵活。要说省心的话,TGI可能稍微占优,坑少一点。不过你这种小规模场景,真没必要纠结太久,挑一个先跑起来,不合适再换也不费劲。