最近在试着微调一个7B的chat模型,用的LoRA,单卡A100 40G。训练的时候loss降得挺正常,但一跑推理就报CUDA OOM,而且加载模型的时候就要吃掉快25G显存。我试了FP16和int8量化,感觉也没好多少。看别人的部署教程好像很轻松,是不是我的batch size或者max length设置有问题?还是说7B模型本来就这德行,得用vLLM或者什么其他框架才行?求有经验的大佬指点一下,给个大致的方向就行,谢谢了!
楼主
2026-08-17
部署7B大模型微调后显存爆了,求大佬看看是不是我配置有问题?
请 登录 后发表回复
全部回复
共 62 条
2楼
5天前
学到了,感谢分享!
3楼
7小时前
加载就吃25G挺正常的,7B的fp16权重本身就14G左右,加上CUDA context和框架开销差不多这个数。你推理OOM大概率是max length开太大或者batch没设对,KV cache在长序列下涨得很凶。建议换vLLM或者TGI试试,PagedAttention对显存管理好很多,同样的卡能跑大不少。LoRA微调完记得把adapter merge进去再部署,不然推理时多一层开销也容易出问题。