在今天凌晨的Google I/0 2018大會上,谷歌發(fā)布了Android P新系統(tǒng)以及一大波新應用,與硬件相關的并不多,主要的就是谷歌的新一代TPU處理器——TPU 3.0??紤]到谷歌很多應用越來越依賴AI人工智能,TPU 3.0的性能相比目前的TPU 2.0有8倍提升,可達10億億次,計算規(guī)模堪比TOP500超算冠軍,所以散熱要求也很高,谷歌直接上了水冷系統(tǒng)。
TPU全名為Tensor Processing Unit(張量處理單元),是谷歌研發(fā)的一種神經(jīng)網(wǎng)絡訓練的處理器,主要用于深度學習、AI運算,其他公司也有各自研發(fā)的AI處理器,叫法不一而已,比如有的叫做NPU神經(jīng)網(wǎng)絡處理器等等,主要用途都差不多。
TPU的特點是AI運算性能比傳統(tǒng)CPU、GPU以及FPGA處理器都要強,同時功耗要低,能效比很高,去年IO大會上發(fā)布了TPU 2.0,谷歌就宣稱TPU的能效可達GPU的30-80倍,后來還招致NVIDIA的不滿。
今年的TPU 3.0在性能上更上一層樓,谷歌CPU皮查伊在發(fā)布會上表示TPU 3.0的性能達到了TPU 2.0的8倍多,性能可達100PFLOPS,也就是10億億次水平。只看運算能力的話,10億億次的TPU 3.0堪比TOP500超算冠軍——中國神威·太湖之光,其Linpack浮點性能就是9.3億億次,不過超算跟TPU 3.0沒啥可比性,一個是專用處理器,一個更加通用。
TPU 3.0急劇提升的性能對系統(tǒng)設計也提出了更高要求,散熱方式已經(jīng)不可能采用TPU 2.0的風冷設計了,谷歌設計了一套水冷系統(tǒng)來給TPU 3.0降溫。
TPU 3.0除了谷歌自己用之外,未來數(shù)月內(nèi)還會開放給其他公司及開發(fā)人員,TPU云端平臺每小時價格6.5美元即可租用TPU 3.0來訓練自己的神經(jīng)網(wǎng)絡模型。
本文圖片來自Hardware.info。
TPU 2.0采用風冷散熱方式