< 返回新闻公共列表

deepseek所需显卡内存怎么计算?

发布时间:2025-04-21 23:35:45

DeepSeek模型所需显卡内存的计算方法如下:


1、模型参数部分

模型参数部分显存计算公式为:


deepseek所需显卡内存怎么计算?.png


FP16/BF16:每个参数占用2字节

INT8:每个参数占用1字节

INT4:每个参数占用0.5字节


2、激活参数部分

激活参数部分显存计算公式为:


deepseek所需显卡内存怎么计算?1.png

 

3、KV Cache部分

KV Cache部分显存计算公式为:

\text{KV Cache显存(GB)} = \frac{\text{并发数} \times (\text{输入Token数} + \text{输出Token数}) \times 2 \times \text{层数} \times \text{hidden_size} \times \text{Sizeof(精度系数)}}{10^9}


4、总显存

总显存为以上三部分之和:

总显存(GB)=模型参数显存+激活参数显存+KV Cache显存


5、安全系数

为了预留缓存和系统开销,建议乘以一个安全系数(1.2-1.5):

总显存(GB)=总显存×安全系数


6、优化策略

量化技术:使用INT8或INT4量化可显著降低显存需求。

异构计算:通过将稀疏MoE矩阵卸载到CPU内存,仅保留稠密部分在GPU显存中。

算子优化:使用Marlin算子加速量化计算,结合CUDA Graph减少显存碎片,提升利用率。


7、实际计算示例

以DeepSeek-R1 671B模型为例,假设batch size=30,isl=2048,out=2048,num_layers=61,hidden_size=7168,激活参数量37B:

模型参数显存:671B×2字节 = 1342 GB

激活参数显存:37B×2字节 = 74 GB

KV Cache显存:30×(2048 + 2048)×2×61×7168×2字节= 808.08 GB

总显存:1342 + 74 + 808.08 = 2224.08 GB

考虑安全系数1.2,总显存需求约为2668.9 GB。


8、量化后的显存需求

INT8量化:671B模型显存需求约为671 GB(参数)+约5.3 GB(KV Cache) ≈ 676.3 GB。

INT4量化:671B模型显存需求约为335.5 GB(参数)+约5.3 GB(KV Cache) ≈ 340.8 GB。


通过以上步骤和公式,可以准确计算DeepSeek模型在不同配置下的显卡内存需求。


/template/Home/Zkeys724/PC/Static