Google I/O 2018开发者大会期间,Google正式发布了第三代AI人工智能/机器学习专用处理器TPU 3.0。
TPU也就是Tensor Processor Unit,Google为机器学习定制的专用芯片(ASIC),专为Google的深度学习框架TensorFlow而设计。
相比于传统GPU图形芯片,它使用8位低精度计算以节省晶体管,对精度影响很小但可以大幅节约功耗、加快速度,同时还有脉动阵列设计,优化矩阵乘法与卷积运算,并使用更大的偏上内存,减少对系统内存的依赖。
TPU 3.0的计算能力最高可达100PFlops(每秒1000万亿次浮点计算),是去年第二代的8倍还多。
它还第一次使用了液冷散热,可以更高效地服务数据中心,便于定制硬件方案。
更多技术细节待公布……
谷歌CEO桑达尔·皮查伊(Sundar Pichai)称,当人们使用大量第三代TPU时,它可能会创造庞大计算能力。“每个芯片群(pods,包含多个TPU 3.0芯片)的性能比去年版本强大8倍以上,计算能力远超100 petaflops(每秒千万亿次浮点运算),”皮查伊称。作为对比,包含16颗英伟达最新GPU的芯片群的计算能力为2 petaflops。
TPU 3.0采用液体冷却(liquid-cooled)技术。这种技术有时被用于高性能计算芯片或者一些用于PC中的性能导向型芯片。
谷歌在去年宣布的第二代TPU已经展示出了不错的性能。近几个月的测试结果显示,在特定情况下,第二代TPU展现出的性能好于现有GPU,尽管它仍存在一定局面性,例如缺乏对PyTorch AI软件框架的支持。PyTorch获得了Facebook的支持,该开源社区正在努力改变这一点。
领取专属 10元无门槛券
私享最新 技术干货