比CPU/GPU快30倍!Google TPU机器学习芯片揭秘
昨日,Google资深硬件工程师Norman Jouppi刊文表示,Google的专用机器学习芯片TPU处理速度要比GPU和CPU快15-30倍(和TPU对比的是IntelHaswell CPU以及NVIDIA Tesla K80 GPU),而在能效上,TPU更是提升了30到80倍。
从这次发布的测试结果来看,TPU似乎已经超出了业界的预期,但是藏在这一芯片背后的内部架构究竟有什么秘密呢?我们从Jouppi此前发布的论文当中,可以找到答案。
据雷锋网了解,早在四年前,Google内部就开始使用消耗大量计算资源的深度学习模型。这对CPU、GPU组合而言是一个巨大的挑战,Google深知如果基于现有硬件,他们将不得不将数据中心数量翻一番来支持这些复杂的计算任务。
所以Google开始研发一种新的架构,Jouppi称之为“下一个平台”。Jouppi曾是MIPS处理器的首席架构师之一,他开创了内存系统中的新技术。三年前他加入Google的时候,公司上下正在用CPU、GPU混合架构上来进行深度学习的训练。
Jouppi表示,Google的硬件工程团队在转向定制ASIC之前,早期还曾用FPGA来解决廉价、高效和高性能推理的问题。但他指出,FPGA的性能和每瓦性能相比ASIC都有很大的差距。
他解释说,TPU可以像CPU或GPU一样可编程,它可以在不同的网络(卷积神经网络,LSTM模型和大规模完全连接的模型)上执行CISC指令,而不是为某个专用的神经网络模型设计的。
一言以蔽之,TPU兼具了CPU和ASIC的有点,它不仅是可编程的,而且比CPU、GPU和FPGA拥有更高的效率和更低的能耗。
TPU的内部架构
该图显示了TPU上的内部结构,除了外挂的DDR3内存,左侧是主机界面。指令从主机发送到队列中(没有循环)。这些激活控制逻辑可以根据指令多次运行相同的指令。
TPU并非一款复杂的硬件,它看起来像是雷达应用的信号处理引擎,而不是标准的X86衍生架构。
Jouppi说,尽管它有众多的矩阵乘法单元,但是它比GPU更精于浮点单元的协处理。另外,需要注意的是,TPU没有任何存储的程序,它可以直接从主机发送指令。
TPU上的DRAM作为一个单元并行运行,因为需要获取更多的权重以馈送到矩阵乘法单元(算下来,吞吐量达到了64,000)。Jouppi并没有提到是他们是如何缩放并行结构的,但他表示,使用的主机软件加速器都将成为瓶颈。
从第一张图片可以看出,TPU有两个内存单元,以及一个用于模型中参数的外部DDR3 DRAM。参数进来后,可从顶部加载到矩阵乘法单元中。同时,可以从左边加载激活(或从“神经元”输出)。那些以收缩的方式进入矩阵单元以产生矩阵乘法,它可以在每个周期中进行64,000次累加。
毋庸置疑,Google可能使用了一些新的技巧和技术来加快TPU的性能和效率。例如,使用高带宽内存或混合3D内存。然而,Google的问题在于保持分布式硬件的一致性。
TPU对比Haswell处理器
在和Intel“Haswell”Xeon E5 v3处理器来的对比中,我们可以看到,TPU各方面的表现都要强于前者。
在Google的测试中,使用64位浮点数学运算器的18核心运行在2.3 GHz的Haswell Xeon E5-2699 v3处理器能够处理每秒1.3 TOPS的运算,并提供51GB/秒的内存带宽;Haswell芯片功耗为145瓦,其系统(拥有256 GB内存)满载时消耗455瓦特。
相比之下,TPU使用8位整数数学运算器,拥有256GB的主机内存以及32GB的内存,能够实现34GB/秒的内存带宽,处理速度高达92 TOPS ,这比Haswell提升了71倍,此外,TPU服务器的热功率只有384瓦。
除此之外,Google还测试了CPU、GPU和TPU处理不同批量大小的每秒推断的吞吐量。
如上图所示,在小批量任务中(16),Haswell CPU的响应时间接近7毫秒,其每秒提供5482次推断(IPS),其可以实现的最大批量任务(64)每秒则可以完成13194次推断,但其响应时间为21.3毫秒。相比之下,TPU可以做到批量大小为200,而响应时间低于7毫秒,并提供225000个IPS运行推理基准,是其峰值性能的80%,当批量大小为250,响应时间为10毫秒。
不过需要注意的是,Google所测试的Haswell Xeon处理器似乎也不能完全说明问题,IntelBroadwell Xeon E5 v4处理器和最新的“Skylake”Xeon E5,每核心时钟(IPC)的指令比这款处理器提升了约5%。在Skylake是28核,而Haswell为18核,所以Xeon的总体吞吐量可能会上升80%。当然,这样的提升与TPU相比仍有差距。
最后雷锋网需要强调的是,TPU是一个推理芯片,它并非是要取代GPU,可以确定的是,TPU与CPU一起使用对训练分析更加有益,但对于CPU制造商而言,如何研发出像ASIC一样兼顾性能和能效的芯片是现在以及未来要做的。
Jouppi表示GoogleTPU已经开始出货,而Intel这些芯片商也将面临更大的挑战。
-
安卓QQ国际版更新v5.2.0:优化语音聊天功能
适用于Android系统的QQ国际版v5 2 0已于日前正式发布,详细的版本号为v5 2 0 6068,此次的更新虽未带来新的功能,却进行了一些修复和优化。
详情2017-08-03 10:05:39责编:$(username)} 来源:驱动管家 人气:0 -
惊喜!Surface Laptop处理器确认 完虐MacBook
微软正式发布了Surface Laptop,这是一款主打轻薄的笔记本电脑,和Surface Book不同的是它并未采用二合一设计,所以定位并不冲突,而且价格更便宜。
详情2017-08-03 17:02:58责编:$(username)} 来源:驱动管家 人气:0 -
微信小程序越发开放:可共享到微信群
经过一段时间的沉默,微信小程序再次迎来新功能,今天正式开放了群相关能力。
详情2017-08-03 16:12:40责编:$(username)} 来源:驱动管家 人气:0 -
Uber公司2019年会上市!新任CEO科斯罗萨西称董事会支持IPO
Uber公司2019年会上市,这是Uber新任CEO达拉·科斯罗萨西在11月9号接受采访时说的。对于这个决定,科斯罗萨西称,虽然来自沙特和日本的大股东并没有逼迫Uber上市,但是公司的创始人和董事会支持IPO,即首次公开招
详情2017-11-10 12:50:03责编:$(username)} 来源:驱动管家 人气: -
微星主板捆绑Intel傲腾缓存SSD:提速14倍!
Intel Optane傲腾存储技术已经先后诞生了两种产品样式,分别是面向数据中心的PCI-E SSD、面向消费市场的Caches SSD,后者用于搭配传统机械硬盘,起加速作用,实测表明效果还是很明显的。
详情2017-08-03 17:39:16责编:$(username)} 来源:驱动管家 人气:1 -
BIOS更新就绪!AMD Ryzen再爆发:内存兼容看齐Intel
此前,AMD曾透露,安排了超过300人的工程专家团队来为Ryzen“护航”,现在他们将拿出最新成果,即新的AGESA(微处理器封装架构)。
详情2017-08-03 16:35:44责编:$(username)} 来源:驱动管家 人气:5 -
腾讯的电信网络诈骗报告出炉了 小心短信诈骗
网络诈骗一直以来都是骗取钱财的主要手段,不知道有多少人的血汗钱被骗走,不知道有多少家庭因此变得支离破碎,还有不少人受不了打击自杀。骗子自古以来都是人们最厌恶的角色。这些网络骗子被抓到坐牢都是便宜他
详情2017-11-13 17:57:02责编:$(username)} 来源:驱动管家 人气: -
16年前的今天 Windows XP诞生 从此影响一代人
2001年10月25日,微软公司开发的Windows XP操作系统正式面向全球发布。也许连微软也没有预料到,这款Windows XP系统能够统治全球桌面系统市场十几年。Windows XP是第一个采用NT内核的Windows消费者版本,这个
详情2017-10-25 20:11:11责编:$(username)} 来源:驱动管家 人气:0 -
好用的浏览器:百分浏览器2.6.5发布
Cent Browser,又称百分浏览器,是一款由百分工作室(Cent Studio)采用谷歌的开源Chromium开发者引擎系统精心重编译而成,并附加诸多实用功能的增强版网页浏览器。
详情2017-08-03 10:22:18责编:$(username)} 来源:驱动管家 人气:4 -
特斯拉的电动汽车引入“冷静模式”, 不失速度却更平滑温和
小时候我们看过一部动画片《大头文字D》,剧中讲述的是飙车的故事。隔着屏幕的我们看着里面的车行驶非同一般的速度,心里十分感觉十分刺激和紧张,但是看到这疯狂的车速,真叫我们坐在车上是绝对不敢的。因为这样
详情2017-11-11 16:01:56责编:$(username)} 来源:驱动管家 人气:
- Win8.1开机的时候老是弹出syslevelup提示框怎么办?
- excel如何合并单元格?excel合并单元格的方法
- 在神舟台式电脑上激活ESET NOD32防病毒软件需要注意什么?
- 决心担任好惠普CEO的惠特曼突然宣布离职导致惠普股价大跌
- 爱普生针式打印机不能打印怎么解决?
- excel模板的制作方法是什么?怎么制作excel模板
- 中了AUTORUN 病毒的东芝电脑省电程序不停弹出无法关闭
- 苹果把零售店开到韩国 韩媒颇为不满
- 你适合买波轮洗衣机还是滚筒洗衣机?这里有二者长短处对比
- 教你一招解决电脑“无法找到打印机驱动程序包要求的核心驱动”的问题
- 电脑的网卡驱动装不上?你肯定没试过这招
- Excel if函数怎么用?关于excel if函数的用法介绍
- 如何处理新建宽带连接时神舟一体电脑无法输入用户名和密码的情况?
- 东芝笔记本时不时断开网络,需要频繁点击保持连接
- 腾讯市值超5000亿美元,其CEO马化腾身家483亿美元
- 又降价了!感恩节京东手机低至5折,劵多多
- 关于电脑驱动的这些问题你都知道多少?
- excel高级筛选功能的使用方法是什么?
- 什么原因导致华硕A46CM系列电脑休眠后无法唤醒?解决措施又是什么?
- 机器人做医生?机器人在国家医师执照考试中取得高及格线96分的分数