7 月 28 日消息,Meta 發(fā)布的一份研究報告顯示,其用于訓(xùn)練 4050 億參數(shù)模型 Llama 3 的 16384 個英偉達 H100 顯卡集群在 54 天內(nèi)出現(xiàn)了 419 次意外故障,平均每三小時就有一次。其中,一半以上的故障是由顯卡或其搭載的高帶寬內(nèi)存(HBM3)引起的。
由于系統(tǒng)規(guī)模巨大且任務(wù)高度同步,單個顯卡故障可能導(dǎo)致整個訓(xùn)練任務(wù)中斷,需要重新開始。盡管如此,Meta 團隊還是保持了 90% 以上的有效訓(xùn)練時間。
在為期 54 天的預(yù)訓(xùn)練中,共出現(xiàn)了 466 次工作中斷,其中 47 次是計劃中斷,419 次是意外中斷。計劃內(nèi)的中斷是由于自動化維護造成的,而意外的中斷則主要源于硬件問題。 GPU 問題是導(dǎo)致故障的主要原因,占意外中斷的 58.7%。其中只有三起事件需要大量人工干預(yù),其余的由自動化管理。
在 419 個意外中斷中,148 個(30.1%)是由各種 GPU 故障(包括 NVLink 故障)引起的,而 72 個(17.2%)是由 GPU 的 HBM3 內(nèi)存故障引起的。有趣的是,54 天內(nèi)只有兩個 CPU 發(fā)生故障。41.3% 的意外中斷是由多種因素造成的,包括軟件錯誤、網(wǎng)絡(luò)電纜和網(wǎng)絡(luò)適配器。
為提高效率,Meta 團隊開發(fā)了一系列工具和優(yōu)化策略,包括縮短任務(wù)啟動和檢查點時間、利用 PyTorch 的 NCCL 飛行記錄器診斷性能問題、識別拖后顯卡等。此外,Meta 還關(guān)注到了環(huán)境因素的影響,如午間溫度波動對 GPU 性能的輕微影響,以及巨量 GPU 同時運行對數(shù)據(jù)中心電網(wǎng)的巨大壓力。
然而,隨著人工智能模型參數(shù)量的不斷增加,所需的計算資源也隨之?dāng)U大。以 xAI 計劃中的 10 萬塊 H100 顯卡集群為例,故障率可能會成倍增長,給未來的 AI 訓(xùn)練帶來更大的挑戰(zhàn)。