上文用簡(jiǎn)單的小學(xué)數(shù)學(xué)算了一下Alexnet的參數(shù)說(shuō)需要的內(nèi)存空間，但對(duì)于運(yùn)行的神經(jīng)網(wǎng)絡(luò)，還有一個(gè)運(yùn)行時(shí)的資源的問題。在github上的convnet-burden上有一個(gè)feature memory[1]的概念，這個(gè)和輸入的圖片的大小和運(yùn)算的batch的size 都有關(guān)。

因此，Nvida的GPU上的HBM和GDDR對(duì)于大部分神經(jīng)網(wǎng)絡(luò)的煉丹師都是非常重要，能夠在一個(gè)GPU的內(nèi)存里完成模型的運(yùn)算而不用考慮換進(jìn)換出是大有裨益的。

總體來(lái)說(shuō)，這個(gè)統(tǒng)計(jì)還是很直觀的[2]，就是網(wǎng)絡(luò)模型越復(fù)雜，參數(shù)的規(guī)模越大，資源的占用也就越多，對(duì)GPU的整體內(nèi)存占用也是越多。因此如何在有限的GPU上完成模型的訓(xùn)練也成了一個(gè)非常有用的技巧。

在我們考慮計(jì)算對(duì)于內(nèi)存帶寬的需求之前，我們需要復(fù)習(xí)一下作為一個(gè)神經(jīng)網(wǎng)絡(luò)，每一層對(duì)于計(jì)算的需求，這個(gè)還是可以用小學(xué)數(shù)學(xué)搞定的東西。還是用標(biāo)準(zhǔn)的Alexnet為例。

對(duì)于計(jì)算來(lái)講，卷積層應(yīng)該是主要，對(duì)于Pooling來(lái)講，應(yīng)該是沒有的，對(duì)于FC來(lái)講，也是比較簡(jiǎn)單的。基本上是乘法.

Conv Layer的計(jì)算復(fù)雜度：

1. 當(dāng)前的層的圖片的width

2. 當(dāng)前的層的圖片的height

3.上一層的深度

4。當(dāng)前層的深度

5. 當(dāng)前kernel的大小

的乘積就是這一層卷積的計(jì)算復(fù)雜度。以Alexnet的conv1為例：

Conv-1:第一層的卷積有96個(gè)kernel。kernel的大小是11X11，卷積的stride是4，padding是0

當(dāng)前的輸出的是55X55，上一層的input的深度是3， 當(dāng)前的kernel是11X11，當(dāng)前的深度是96.因此

55X55X3X11X11X96=105,415,200次MAC的計(jì)算。

對(duì)于Alexnet來(lái)講，需要注意的是conv2，4 ，5三個(gè)層的計(jì)算沒有和上一層直接跨GPU，因此需要的計(jì)算規(guī)模上/2.

對(duì)于FC來(lái)講，比較粗略的計(jì)算就是 輸入和當(dāng)前層的規(guī)模的乘積。

MaxPool-3：第五層卷積的最大值，Pooling是3X3， stride是2，

FC-1：第一個(gè)全連接層，有4096個(gè)神經(jīng)元

因此FC-1 的計(jì)算就是：

6X6X256X4096=37,748,736

因此，總結(jié)一下之前的參數(shù)信息和計(jì)算量， Alexnet的圖如下：

這個(gè)通過統(tǒng)計(jì)每一層的計(jì)算的復(fù)雜度，就可以得到整個(gè)網(wǎng)絡(luò)的計(jì)算復(fù)雜度，也就是訓(xùn)練一次網(wǎng)絡(luò)需要多少的MACC資源。對(duì)于alexnet 來(lái)講就是：724,406,816 次操作。

這個(gè)時(shí)候，有一個(gè)關(guān)鍵的信息就出來(lái)了。就是芯片的能力，大家都是用TFLOPs來(lái)表示芯片的浮點(diǎn)處理能力。對(duì)于Nvida的芯片，有了TFLOPS，有個(gè)一個(gè)網(wǎng)絡(luò)需要的計(jì)算量，我們就可以很快計(jì)算出每一層計(jì)算需要的時(shí)間了。

對(duì)于Alexnet 的conv1 來(lái)講，在Nivida 最新的V100的120TFLOPs的GPU上，進(jìn)行訓(xùn)練的執(zhí)行時(shí)間差不多是105,415,200X2/(120X1,000,000,000,000), 約等于1.75us (微秒）。

對(duì)于Pooling這一層來(lái)講，因?yàn)闆]有MACC的計(jì)算量，但是因?yàn)橐狹ax Pooling，也需要大小比較的計(jì)算。因此，它的計(jì)算基本就是算是數(shù)據(jù)讀取。因此它的數(shù)據(jù)讀取是 conv-1 的55X55X96=290,440. 因此在同樣的GPU下，它的執(zhí)行時(shí)間就是 2.42ns.

好了，有了計(jì)算時(shí)間，現(xiàn)在需要來(lái)計(jì)算數(shù)據(jù)量了。對(duì)于Conv1來(lái)講，它包含了對(duì)一下數(shù)據(jù)的讀寫：

對(duì)于輸入數(shù)據(jù)的讀取 227X227X3 =154,587

對(duì)于輸出數(shù)據(jù)的寫入55x55x96=290,400

對(duì)于參數(shù)的讀取34848+96=34944

因此，就可以算出對(duì)于120TFLOP的GPU的要求：因?yàn)樵诤芏?a target="_blank">ASIC芯片中，輸入輸出可能在DDR中，但是參數(shù)可能放在SRAM中，因此我們就分開計(jì)算了。

對(duì)于輸入數(shù)據(jù)的讀取 （154,587/1.75）X1000,000X4=351.95GB/s

對(duì)于輸出數(shù)據(jù)的寫入 （290,400/1.75）X1000,000X4=661.16GB/s

對(duì)于參數(shù)的讀取 (34944 /1.75)X1,000,000X4=79.34GB/s

提個(gè)醒，我們現(xiàn)在的PC服務(wù)器上性能最高的DDR4的帶寬基本上在19GB/s左右。看到壓力了吧?，F(xiàn)在的memory連很慢的CPU都跟不上。更何況老黃家的核彈。