返回文章
硬件与成本3 分钟阅读

7B 模型需要多少内存?先算权重,再留余量

看到“7B、4-bit”就判断一张显卡能运行模型,容易漏算其他内存。权重大小可以先手算,但它只是整套推理程序的一部分。把理论大小和运行峰值分开,硬件选择会更清楚。

无品牌计算芯片与分层电路、内存模块的概念插画。
AI 生成的编辑插画用于表现文章主题,并非产品截图、研究结果或原始资料照片。
三个月免费开放 · 至 2027 年 1 月 10 日(日本时间),所有文章均可阅读全文。

先用一个简化公式估算权重:

权重字节数约等于参数数量 × 每个参数的位数 ÷ 8。

把 7B 理解为约七十亿个参数,得到以下理论值:

权重精度理论权重大小,十进制 GB
16-bit约 14GB
8-bit约 7GB
4-bit约 3.5GB

这些数字是算术估算,不是某个模型的实测显存。真实实现还包括量化元数据、未量化的层、运行缓冲区、激活值和 KV cache;GB 与 GiB 的换算也会造成数字差异。

Hugging Face 的文档介绍了用 bitsandbytes 加载较低精度权重的方法,以及相关计算类型和实现条件。实际收益需要结合模型与运行方式判断。参考文档 ↗

如果目标是设备选型,建议先固定三个条件:模型与量化版本、最长输入、同时处理的请求数。然后测量在这三个条件下的峰值内存、首字延迟和生成速度。单看参数量,会遗漏你的实际使用方式。

视觉札记
半透明文件切片经过检查门,连接计算芯片与独立内存模块,示意本地推理的不同组成部分。
AI 生成的概念插画看完整的内存预算

试一试

列出准备运行的模型,分别算一次理论权重。再给测试表增加输入长度、并发、峰值内存和速度四列。尚未跑过的模型留空,不把理论估算填成实测值。

如果只是处理每天一篇文章,比较远程 API 与本地设备的总成本时,还应计入购置、闲置、电力与维护。

把原理,切开来看

看完整的内存预算

  1. 模型权重

    实际加载的权重

  2. KV 缓存

    上下文与并发请求

  3. 运行开销

    缓冲区与其他分配

编辑示意图这里只区分组成部分,不表示实测占比。激活参数量描述计算量,不能单独告诉你需要多少内存。

读懂之后,再动手试一次。

看看下一篇
这篇文章对你有帮助吗?

无需登录,同一浏览器在不同语言中只计一次。

正在读取点赞…

原始参考资料

这些资料用于核查与继续学习,来源不代表对本刊的背书。

huggingface.co参考原文 1