AI硬體與軟體堆疊

若以FP16(2位元組)持有參數量70億(7B)的模型權重,權重的記憶體大約是多少?

  1. A約14 GB
  2. B約7 GB
  3. C約28 GB
查看答案與解析

正確答案:A. 約14 GB

權重的記憶體可由參數數量×每個參數的位元組數求得。70億×2位元組=140億位元組,約為14 GB。推論時,除此之外還需要中間結果,以及大型語言模型的KV快取所用的記憶體。要判斷模型能在哪種裝置或GPU上執行,首先就以這個計算來預估。

  • A. 正確。70億×2位元組。
  • B. 這是每個參數1位元組(INT8)時的值。
  • C. 這是FP32(4位元組)時的值。
練習這個領域 前往領域頁面

AI硬體與軟體堆疊的其他題目

  1. 將上一題的7B模型從FP16(約14 GB)量化為4位元整數(INT4),權重的記憶體大約是多少?
  2. 計算M×K矩陣與K×N矩陣的乘積時,浮點運算的次數(乘法與加法分開計算)大約是多少?
  3. 計算N×N的FP16(2位元組)矩陣相乘,兩個輸入各讀取1次,結果寫入1次。運算強度(每位元組的運算數)大約是多少?
  4. 以尖峰效能100 TFLOPS、記憶體頻寬2 TB/s的加速器執行運算強度為10 FLOP/位元組的處理。以Roofline模型估算,可達成的效能為何?
  5. 輸入只有1筆(批次大小為1)時,全連接層的計算(權重矩陣與向量的乘積)容易受記憶體限制,其原因下列何者正確?