设置
  • 日夜间
    随系统
    浅色
    深色
  • 主题色

训练 GPT3-175B 模型最高快 180%,谷歌宣布云端 AI 芯片 TPU v5p

2023/12/7 12:35:13 来源:IT之家 作者:故渊 责编:故渊
感谢IT之家网友 肖战割割溯波 的线索投递!

IT之家 12 月 7 日消息,谷歌在推出全新大语言模型 Gemini 1.0 之外,还宣布了增强版 Tensor Processing Unit(TPU)芯片,最新型号为 TPU v5p。

谷歌目前已经在 YouTube、Gmail、Google Maps、Google Play 和 Android 等产品服务中使用 TPU 芯片,最新版本是谷歌迄今为止功能最强大、可扩展性最强和最灵活的 AI 加速器芯片。

IT之家注:在性能方面,谷歌 TPU v5p 在 bfloat16 精度下,可以实现 459 teraFLOPS;在 Int8 精度下,可以实现 918 teraOPS

谷歌 TPU v5p 配备 95GB 的 HBM3 内存,内存带宽为 2.76TB / sec,每个 Pod 最多有 8960 个加速核心,并自研搭载 600GB/sec 芯片互联主控,可以更快、更准确地训练 AI 模型。

与 TPU v4 相比,新发布的 TPU v5p 具有两倍的 FLOPS(每秒浮点运算)和三倍的高内存带宽提升。

谷歌表示在 BF16 精度下,训练 OpenAI 拥有 1750 亿参数的 GPT3,比此前的 TPU v4 AI 加速器芯片快 90%;如果在 Int8 精度下可以达到 180%。

每个 TPU v5p 加速器的运行费用为每小时 4.20 美元,这比 TPU v4(每小时运行费用为 3.22 美元)和 TPU v5e(每小时运行费用为 1.20 美元)贵一点。

广告声明:文内含有的对外跳转链接(包括不限于超链接、二维码、口令等形式),用于传递更多信息,节省甄选时间,结果仅供参考,IT之家所有文章均包含本声明。

相关文章

关键词:谷歌TPU

软媒旗下网站: IT之家 辣品 - 超值导购,优惠券 最会买 - 返利返现优惠券 iPhone之家 Win7之家 Win10之家 Win11之家

软媒旗下软件: 魔方 酷点桌面 Win7优化大师 Win10优化大师 软媒手机APP应用