使用 DeepChat 的 3 步訓(xùn)練示例，對捷智算平臺提供的 NVIDIA H100 SXM5 與 NVIDIA A100 SXM4 進(jìn)行 GPU 基準(zhǔn)測試。
目標(biāo)
我們使用DeepChat 的 3 步訓(xùn)練示例，對捷智算平臺提供的NVIDIA H100 SXM5 實(shí)例（由 NVIDIA H100 Tensor Core GPU 提供支持）進(jìn)行了激動人心的 GPU基準(zhǔn)測試。我們將性能與參考 NVIDIA A100 SXM4 Tensor Core 系統(tǒng)進(jìn)行比較，并在 128 臺服務(wù)器上的 1,024 個 GPU 上對其可擴(kuò)展性進(jìn)行壓力測試。
準(zhǔn)備事項(xiàng)
每臺服務(wù)器配備 8 個 NVIDIA H100 SXM5 GPU 和 8 個 400Gb/s NDR InfiniBand 鏈路。這相當(dāng)于 640GB 的 GPU 內(nèi)存和 3200Gb/s 的節(jié)點(diǎn)間帶寬。
利用完全無阻塞的軌道優(yōu)化網(wǎng)絡(luò)拓?fù)洌覀冏畲笙薅鹊靥岣吡巳阅懿p少了網(wǎng)絡(luò)沖突，確保服務(wù)器之間的InfiniBand 性能大于 750Gbit/s ，這通過一對 InfiniBand 端口之間的雙向 ib_write_bw 測試來測量。
所有服務(wù)器均預(yù)裝了 Lambda Stack、InfiniBand 驅(qū)動程序和 deepspeed 0.10.0，并同步到共享存儲以用于訓(xùn)練數(shù)據(jù)和預(yù)訓(xùn)練權(quán)重。
主要結(jié)果
捷智算平臺的 NVIDIA H100 SXM5 與 NVIDIA A100 SXM4 實(shí)例在 FP16 中的 3 步強(qiáng)化學(xué)習(xí)人類反饋 (RLHF) 管道上的正面比較顯示：
步驟 1（OPT-13B Zero3）：NVIDIA H100 速度提高 2.8 倍。
步驟 2（OPT-350M Zero0）：NVIDIA H100 獲得 2.5 倍速度優(yōu)勢。
步驟 3（OPT-13B Zero3 加 OPT-350M Zero0）：NVIDIA H100 以 3.1 倍的速度提升遙遙領(lǐng)先。
測試分布式訓(xùn)練可擴(kuò)展性：
大型模型（OPT-13B）和更大的批次（16 個樣本/GPU）導(dǎo)致 128 臺服務(wù)器的吞吐量達(dá)到 127.51 倍。 
較小的模型（OPT-350M）和較小的批次（4 個樣本/GPU）仍然令人印象深刻，128 臺服務(wù)器的吞吐量達(dá)到 112.79 倍。
結(jié)論
與 NVIDIA A100 SXM4 系統(tǒng)相比，NVIDIA H100 SXM5 系統(tǒng)上的 DeepSpeed 訓(xùn)練速度提高了 2.5 倍至 3.1 倍。捷智算平臺配備 80GB NVIDIA H100 SXM5 GPU、NIC 與 GPU 比率為 1:1 的 InfiniBand 連接以及軌道優(yōu)化網(wǎng)絡(luò)。它們可以在數(shù)千個 GPU 上提供前所未有的性能和可擴(kuò)展性。
