单个Pod性能是世界第一超算的两倍 里程碑式Google TPU v4重磅发布! (pod单是什么意思)

文章编号:44828 资讯动态 2024-12-09 TPUgoogle

Google I/O开发者大会去年因为疫情而取消,今年采取线上形式强势回归。在没有开发者在场的Google园区内,Google CEO桑达尔·皮查伊(Sundar Pichai)宣布推出多项全新技术,除了能够帮助用户实现“空间瞬移”的全息视频聊天技术Project Starling让人耳目一新,还有最新一代AI芯片TPU v4。

“这是我们在Google上部署的最快的系统,对我们来说是一个具有历史意义的里程碑。”皮查伊这样介绍到。

里程碑式Google TPU v4重磅发布! 单个Pod性能是世界第一超算的两倍

最强TPU,速度提升2倍,性能提升10倍

Google官方介绍,在相同的64芯片规模下,不考虑软件带来的改善,TPU v4相较于上一代TPU v3性能平均提升2.7倍。

在实际应用中,TPU v4主要与Pod相连发挥作用,每一个TPU v4 Pod中有4096个TPU v4单芯片,得益于其独特的互连技术,能够将数百个独立的处理器转变为一个系统,互连带宽在规模上是其他任何网络技术的10倍,每一个TPU v4 Pod就能达到1 exaFlOP级的算力,实现每秒10的18次方浮点运算。这甚至是全球最快的超级计算机“富岳”的两倍性能。

“如果现在有1千万人同时使用笔记本电脑,所有这些计算机累加的计算能力,刚好就能够达到1 exaFLOP的算力。而之前要达到1 exaFLOP,可能需要专门定制一个超级计算机。”皮查伊如是说。

今年的MLPerf结果表明,GoogleTPU v4的实力不容小觑,在使用Imagenet数据集的图像分类训练测试(准确度至少75.90%),256 个TPU v4在1.82分钟内完成了这一任务,这几乎与768个Nvidia A100图形卡、192个AMD Epyc 7742内核(1.06分钟)、512个华为AI优化的Ascend910芯片以及128个intel Xeon Platinum 8168内核(1.56分钟)组合在一起的速度一样快。

当负责在大型维基百科语料库上训练基于Transform的阅读理解BERT模型时,TPU v4的得分也很高。使用256个TPU v4进行训练需要1.82分钟,比使用4096 TPU v3进行训练所需的0.39分钟要慢1分多钟。同时,如果想要使用Nvidia的硬件达到0.81分钟的训练时间,需要2048张A100卡和512个AMD Epyc 7742 CPU内核。

Google同样在I/O大会上展示了能够用到TPU v4的具体AI实例,包括能够同时处理网页、图像等多种数据的MUM模型(Multitask Unified Model,多任务统一模型)和专为对话打造的LaMDA都是能够用到TPU v4的场景模型,前者比阅读理解模型BERT强1000倍,适合赋能搜索引擎帮助用户更加高效地得到自己想要的信息,后者则可以与人类进行不间断的对话交流。

这一并不向外出售的TPU,很快将在被部署在Google的数据中心,且90%左右的TPU v4 Pod都将使用绿的能源。 另外,Google也表示,将在今年晚些时候开放给Google Cloud的客户。

Google最早于2016年宣布首款内部定制的AI芯片,区别于训练和部署AI模型的最常见的组合架构,即CPU和GPU组合,第一代TPU在那场世界著名的人机围棋大战助力AlphaGo打败李世石“一战成名”,宣告并不是只有GPU才能做训练和推理。

Google第一代TPU采用28nm工艺制程,功耗大约40w,仅适用于深度学习推理,除了AlphaGo,也用在Google搜索、翻译等机器学习模型中。

2017年5月,Google发布了能够实现机器学习模型训练和推理的TPU v2,达到180TFLOPs浮点运算能力,同时内存带宽也得以提升,比同期推出的CPU AI工作负载提升30倍,比GPU AI工作负载提升15倍,被基于4块TPU v2的AlphaGo击败的世界围棋冠军柯洁最直观地感受了这一切。

2018年5月,Google又发布第三代TPU,性能是上一代TPU的两倍,实现420TFLOPs浮点运算,以及128GB的高带宽内存。

按照一年一次迭代更新的节奏,Google理应在2019年推出第四代TPU,不过这一年的I/O大会上,Google推出的是第二代和第三代TPU Pod,可以配置超过1000颗TPU,大大缩短了在进行复杂的模型训练时所需耗费的时间。

在AI芯片发展史上,无论是从片上内存上,还是从可编程能力来看,Google TPU都是不可多得的技术创新,打破GPU的“垄断”地位,且打开云端AI芯片的新竞争格局。

发展五年的Google TPU在今天依然保持着强劲的竞争力,未来的世界是什么样的?Google TPU已经告诉了我们一小部分答案。

要用硅光单芯片替代3000块TPU的AI初创公司吸引了比尔盖茨的投资

Google 祸不单行

原创文章,未经授权禁止转载。详情见 转载须知 。

全局中部横幅
少数派

少数派致力于更好地运用数字产品或科学方法,帮助用户提升工作效率和生活品质

森林号

森林号汇聚全网大咖,分享优质资讯,了解最新热门信息。让您在创业中更近一步,在文海中找到自己的方向。

东莞数控车床厂家

芮安自成立以来一直致力于数控车床研发,生产,销售为企业导向.企业现拥有多轴车铣复合ES系列,车削中心TC系列及自动化定制RA系列.

1微思网络

厦门微思网络提供思科CCNA,CCNP,CCIE培训,华为HCIA,HCIP,HCIE培训,也提供红帽RHCE培训,OracleOCP培训,VMware培训,是福建省广受认可的IT认证培训中心

企业数字展厅设计

什么是五代数字展馆——数字展厅,以一套空间布局+N套数字内容+定制智控后台,打造视觉、内容符合企业形象的数字展厅,实现多主题内容自由切换,满足多类受众需求。科技主题馆等数字化展馆展厅设计施工一体化解决方案提供商,资质齐全,经验丰富、数字化展馆展厅设计施工一体化解决方案提供商

烈火传奇

百度一下,你就知道

张家界旅游网

张家界信息协会指定权威旅游网站,为海内外旅客提供张家界景点,张家界酒店,张家界民俗,张家界特产,张家界交通,张家界地图,张家界天气,张家界旅游攻略等信息的国际网站!

棉谷义春茶

福贡棉谷春芬种养殖专业合作社成立于2021年12月08日,注册地位于中国云南省怒江傈僳族自治州福贡县匹河乡棉谷村委会.我们是一家专注于茶叶种植和销售的企业.联系电话:18724807086.

中公社区工作者招聘考试网

【中公社区工作者招聘考试网】提供2024年社区工作者考试信息,包括社区工作者考试报名入口,社区工作者考试资料,社区工作者工资待遇,社区工作者培训班等信息。

东阳诚信缝配中心(东阳市涵阳服装机型有限公司)是一家集生产

东阳诚信缝配中心(东阳市涵阳服装机型有限公司)是一家集生产、销售与贸易为一体的缝纫机零配件及整机工贸企业。CHENGXINSEWINGPARTSCENTER

中信所网站

中国科学技术信息研究所,中信所,国家工程图书馆,国家工程技术图书馆,国家工程技术数字图书馆,数字图书馆,国家工程图书馆首页,中国科学技术信息研究所首页,院士著作馆

全局底部横幅