深度学习中GPU和显存分析

Linux查看GPU信息和使用情况

1.查询命令

在linux中，命令行输入nvidia-smi指令即可看到当前nvidia显卡的使用情况

2.参数释义

• Fan：显示风扇转速，数值在0到100%之间，是计算机的期望转速，如果计算机不是通过风扇冷却或者风扇坏了，显示出来就是N/A；
• Temp：显卡内部的温度，单位是摄氏度；
• Perf：表征性能状态，从P0到P12，P0表示最大性能，P12表示状态最小性能；
• Pwr：能耗表示；
• Bus-Id：涉及GPU总线的相关信息；
• Disp.A：是Display Active的意思，表示GPU的显示是否初始化；
• Memory Usage：显存的使用率；
• Volatile GPU-Util：浮动的GPU利用率；
• Compute M：计算模式；

下边的Processes显示每块GPU上每个进程所使用的显存情况。

如果要周期性的动态输出显卡的使用情况，可以在终端输入watch -n 1 nvidia-smi ，-n后面的数字是更新的时间间隔(秒)

 

深度学习中GPU和显存分析（科普）

作者：陈云
链接：https://zhuanlan.zhihu.com/p/31558973
来源：知乎

深度学习最吃机器，耗资源，在本文，我将来科普一下在深度学习中：

• 何为“资源”
• 不同操作都耗费什么资源
• 如何充分的利用有限的资源
• 如何合理选择显卡

并纠正几个误区：

• 显存和GPU等价，使用GPU主要看显存的使用？
• Batch Size 越大，程序越快，而且近似成正比？
• 显存占用越多，程序越快？
• 显存占用大小和batch size大小成正比？

0.预备知识

nvidia-smi是Nvidia显卡命令行管理套件，基于NVML库，旨在管理和监控Nvidia GPU设备。nvidia-smi 命令的输出，其中最重要的两个指标：

• 显存占用
• GPU 利用率

显存占用和 GPU 利用率是两个不一样的东西，显卡是由 GPU 计算单元和显存等组成的，显存和 GPU 的关系有点类似于内存和 CPU 的关系。

这里推荐一个好用的小工具：gpustat, 直接pip install gpustat即可安装，gpustat基于nvidia-smi可以提供更美观简洁的展示

结合 watch 命令，watch --color -n1 gpustat -cpu，可以动态实时监控GPU 的使用情况

显存可以看成是空间，类似于内存。

• 显存用于存放模型和数据等
• 显存越大，所能运行的网络也就越大

GPU 计算单元类似于 CPU 中的核，用来进行数值计算。衡量计算量的单位是 flop： the number of floating-point multiplication-adds，浮点数先乘后加算一个 flop。计算能力越强大，速度越快。衡量计算能力的单位是 flops： 每秒能执行的 flop 数量。

 1*2+3               1 flop
 1*2+3*4+4*5         3 flop

1.显存分析

1.1 存储指标

1Byte = 8 bit
1K = 1024 Byte
1M = 1024 K
1G = 1024 M
1T = 1024 G

10 K = 10*1024 Byte

除了K，M，G，T等之外，我们常用的还有KB、MB，GB，TB，二者有细微的差别：K,，M，G，T是以 1024 为底，而KB、MB，GB，TB以 1000 为底。

1Byte=8bit
1KB=1000Byte
1MB=1000KB
1GB=1000MB
1TB=1000GB
10KB=10000Byte

不过一般来说，在估算显存大小的时候，我们不需要严格的区分这二者。

在深度学习中会用到各种各样的数值类型，数值类型命名规范一般为TypeNum，比如 Int64、Float32、Double64

• Type：有 Int，Float，Double 等
• Num: 一般是 8，16，32，64，128，表示该类型所占据的比特数目

常用的数值类型如下图所示 (int64 准确的说应该是对应c中的long long类型， long类型在32位机器上等效于int32)：

其中 Float32 是在深度学习中最常用的数值类型，称为单精度浮点数，每一个单精度浮点数占用 4Byte 的显存。

举例来说：有一个 1000x1000 的 矩阵，float32，那么占用的显存差不多就是1000x1000x4 Byte = 4MB；32x3x256x256 的四维数组（BxCxHxW）占用显存为：32x3x256x256 / 1024 =24MB

1.2 神经网络显存占用

神经网络模型占用的显存包括：模型自身的参数，模型的输出。举例来说，对于如下图所示的一个全连接网络 (不考虑偏置项 b)

模型的显存占用包括：

• 参数：二维数组 W
• 模型的输出： 二维数组 Y

输入 X 可以看成是上一层的输出，因此把它的显存占用归于上一层。

这么看来显存占用就是W和Y两个数组？

并非如此！！！

下面细细分析。
 

1.2.1 参数的显存占用

只有有参数的层，才会有显存占用。这部份的显存占用和输入无关，模型加载完成之后就会占用。

有参数的层主要包括：

• 卷积
• 全连接
• BatchNorm
• Embedding
  … …

无参数的层：

• 多数的激活层 (Sigmoid/ReLU)
• 池化层
• Dropout
  … …

更具体的来说，模型的参数数目 (这里均不考虑偏置项 b) 为：

• Linear(M->N): 参数数目：M×N

• Conv2d(Cin, Cout, K): 参数数目：Cin × Cout × K × K

• BatchNorm(N): 参数数目： 2N

• Embedding(N,W): 参数数目： N × W

参数占用显存 = 参数数目 ×n

• n = 4 ：float32

• n = 2 : float16

• n = 8 : double64

在 PyTorch 中，当你执行完model=MyGreatModel().cuda()之后就会占用相应的显存，占用的显存大小基本与上述分析的显存差不多（会稍大一些，因为其它开销）。

1.2.2 梯度与动量的显存占用

举例来说， 优化器如果是 SGD：

可以看出来，除了保存 W 之外还要保存对应的梯度，因此显存占用等于参数占用的显存 x2；如果是带 Momentum-SGD

这时候还需要保存动量， 因此显存 x3；如果是 Adam 优化器，动量占用的显存更多，显存 x4；总结一下，模型中与输入无关的显存占用包括：

• 参数 W
• 梯度 dW（一般与参数一样）
• 优化器的动量（普通 SGD 没有动量，momentum-SGD 动量与梯度一样，Adam 优化器动量的数量是梯度的两倍）

1.2.3 输入输出的显存占用

这部份的显存主要看输出的 feature map 的形状：

比如卷积的输入输出满足以下关系：

据此可以计算出每一层输出的 Tensor 的形状，然后就能计算出相应的显存占用。模型输出的显存占用，总结如下：

• 需要计算每一层的 feature map 的形状（多维数组的形状）
• 模型输出的显存占用与 batch size 成正比
• 需要保存输出对应的梯度用以反向传播（链式法则）
• 模型输出不需要存储相应的动量信息（因为不需要执行优化）

对于深度学习中神经网络的显存占用，我们可以得到如下公式：

显存占用 = 模型显存占用 + batch_size × 每个样本的显存占用

可以看出显存不是和 batch-size 简单的成正比，尤其是模型自身比较复杂的情况下：比如全连接很大，Embedding 层很大

另外需要注意：

• 输入（数据，图片）一般不需要计算梯度
• 神经网络的每一层输入输出都需要保存下来，用来反向传播，但是在某些特殊的情况下，我们可以不要保存输入。比如 ReLU，在 PyTorch 中，使用nn.ReLU(inplace = True)能将激活函数 ReLU 的输出直接覆盖保存于模型的输入之中，节省不少显存。感兴趣的读者可以思考一下，这时候是如何反向传播的*（提示：y=relu(x) -> dx = dy.copy();dx[y<=0]=0）*

1.3 节省显存的方法

在深度学习中，一般占用显存最多的是卷积等层的输出，模型参数占用的显存相对较少，而且不太好优化。

节省显存一般有如下方法：

• 降低 batch-size
• 下采样 (NCHW -> (1/4)*NCHW)
• 减少全连接层（一般只留最后一层分类用的全连接层）

2 计算量分析

计算量的定义，之前已经讲过了，计算量越大，操作越费时，运行神经网络花费的时间越多。

2.1 常用操作的计算量

常用的操作计算量如下：

• 全连接层：BxMxN , B 是 batch size，M 是输入形状，N 是输出形状。
• 卷积的计算量:
  
• BatchNorm 计算量我个人估算大概是
  
• 池化的计算量：
  
• ReLU 的计算量： BHWC

2.2 AlexNet 分析

AlexNet 的分析如下图，左边是每一层的参数数目（不是显存占用），右边是消耗的计算资源。这里某些地方的计算结果可能和上面的公式对不上, 这是因为原始的AlexNet实现有点特殊(在多块GPU上实现的)。

可以看出：

• 全连接层占据了绝大多数的参数
• 卷积层的计算量最大

2.3 减少卷积层的计算量

谷歌提出的 MobileNet，利用了一种被称为 DepthWise Convolution 的技术，将神经网络运行速度提升许多，它的核心思想就是把一个卷积操作拆分成两个相对简单的操作的组合。如图所示, 左边是原始卷积操作，右边是两个特殊而又简单的卷积操作的组合（上面类似于池化的操作，但是有权重，下面类似于全连接操作）。

这种操作使得：

• 显存占用变多 (每一步的输出都要保存)
  计算量变少了许多，一般为原来的 10－15%，具体为
  

2.4 常用模型 显存 / 计算复杂度 / 准确率

去年（2016年）一篇论文 (https://arxiv.org/abs/1605.07678) 总结了当时常用模型的各项指标，横座标是计算复杂度（越往右越慢，越耗时），纵座标是准确率（越高越好），圆的面积是参数数量（不是显存占用）。左上角我画了一个红色小圆，那是最理想的模型的的特点：快，效果好，占用显存小。

3 总结

3.1 建议

• 时间更宝贵，尽可能使模型变快（减少 flop）
• 显存占用不是和 batch size 简单成正比，模型自身的参数及其延伸出来的数据也要占据显存
• batch size 越大，速度未必越快。在你充分利用计算资源的时候，加大 batch size 在速度上的提升很有限

尤其是 batch-size，假定 GPU 处理单元已经充分利用的情况下：

• 增大 batch size 能增大速度，但是很有限（主要是并行计算的优化）
• 增大 batch size 能减缓梯度震荡，需要更少的迭代优化次数，收敛的更快，但是每次迭代耗时更长
• 增大 batch size 使得一个 epoch 所能进行的优化次数变少，收敛可能变慢，从而需要更多时间才能收敛（比如 batch_size 变成全部样本数目）

3.2 关于显卡选购

当前（2017年11月）市面上常用的显卡指标如下：
更多显卡的更多指标请参阅
https://en.wikipedia.org/wiki/List_of_Nvidia_graphics_processing_units

显然GTX 1080TI性价比最高，速度超越新 Titan X，价格却便宜很多，显存也只少了 1 个 G（据说故意阉割掉一个 G，不然全面超越了 Titan X 怕激起买 Titan X 人的民愤~）

• K80 性价比很低（速度慢，而且贼贵）
• 注意 GTX TITAN 和 Nvidia TITAN 的区别，别被骗
• tensorcore的性能目前来看还无法全面发挥出来, 这里不考虑. 其它的tesla系列像P100这些企业级的显卡这里不列了,普通消费者不会买, 而且性价比较低(一台DGX 1上百万…)

另外，针对本文，我做了一个 Google 幻灯片：神经网络性能分析，国内用户可以点此下载PPT ；Google 幻灯片格式更好，后者格式可能不太正常。

本文都是针对单机单卡的分析，分布式的情况会和这个有所区别。在分析计算量的时候，只分析了前向传播，反向传播计算量一般会与前向传播有细微的差别。

限于本人水平，文中有疏漏之处，还请指正。