最近在试着把7B的模型部署到本地,显卡是3090(24G),按理说应该够用,但一加载就说CUDA out of memory。我试了加载时传low_cpu_mem_usage=True,也试了把max_memory分配给GPU,但好像没多大用。我看网上说用bitsandbytes做4bit量化,但装了之后又报错说找不到CUDA的setup.py,心态有点崩。就想问问大家,实际部署的时候一般怎么配参数?是不是需要把加载的模型分片到多卡或者用offload?另外,除了量化,还有哪些比较实用的显存优化思路?希望能给个可操作的步骤,谢谢了。
楼主
2026-08-08
部署LLaMA模型一直OOM,大佬们都是怎么优化显存的?
请 登录 后发表回复
全部回复
共 103 条
2楼
2026-09-22
4bit量化确实是最直接的路子,bitsandbytes那个setup.py报错一般是CUDA版本和它编译时不匹配,建议装它官方推荐的那个版本组合,或者干脆用GPTQ/AWQ的预量化模型省心。另外7B在24G上不该OOM,先确认下是不是fp32加载的,用torch_dtype=torch.float16或者bfloat16能省一半。还有个容易被忽略的点,推理时的KV cache也吃显存,把max_new_tokens和batch调小试试。offload到CPU虽然慢但能救急,多卡分片对单卡用户就没必要了。
3楼
5小时前
我之前也遇到过类似问题,后来换了方案。
4楼
2小时前
我之前也遇到过类似问题,后来换了方案。