這是我們 A100 GPU 第二次參與 MLPerf 測試。在另一篇文章中,詳細描述了 GPU、系統、網路與 AI 軟體等方面的進步如何提升運算速度。
舉例來說,我們的工程師找到了一種使用 CUDA Graphs 啟動全神經網路模型的方法,CUDA Graphs 是一套由 NVIDIA CUDA 運作項目及其依賴項目構成的軟體套裝。如此一來便消除過去測試時 CPU 遇到的瓶頸,這些測試將 AI 模型作為許多單獨元件 (kernel) 的鏈來進行釋放。
此外,大規模測試使用 NVIDIA SHARP,該軟體可以整合網路交換器中的多項資料傳輸作業、減少網路流量與等待 CPU 處理的時間。
結合 CUDA Graphs 與 SHARP,使得資料中心可以運用破紀錄的 GPU 數量來進行訓練工作。這是如自然語言處理等許多領域所需要使用到的運算能力,在這些領域裡的 AI 模型規模持續成長,其包含數十億個參數。