谷歌開放第二/三代云端TPU Pod 可配備1000顆TPU
谷歌今天宣布其第二代和第三代Cloud TPU Pod將以beta版的形式提供給公眾。這兩款基於雲計算的超級計算機最多將配置1000顆定制TPU。最新的v3型號性能尤其強大,而且配有液冷系統。每一款設備都可以提供高達每秒100千萬億次浮點數的運算能力。
谷歌表示,按照原始運算能力計算,這種設備可以躋身全球前五大超級計算機行列,但由於TPU的數值精度很低,所以在考慮這個數值時應該打個折扣。
但用戶不必非要使用完整的TPU Pod,谷歌還允許開發者租賃其中的部分算力。但無論如何,這款設備都非常強大,最快可以在兩分鐘時間內訓練一個使用ImageNet數據集的標準ResNet-50圖片分類模型。
TPU v2 Pod最多配備512個核心,運算速度略慢於v3。例如,當使用265個TPU時,v2 Pod可以在11.3分鐘內訓練ResNet-50模型,v3Pod僅需7.1分鐘。使用一個TPU花費的時間為302分鐘。