英伟达H100用11分钟训完GPT-3,PyTorch创始人:不要只看时间

人工智能 新闻
赢了,但没有全赢。

昨天,老黄又「赢麻了」!

为啥呢?原来在最新的 MLPerf 基准测试中,英伟达 H100 GPU 芯片组在以下八项基准测试中全部创下了新纪录,同时成为唯一一个跑完所有测试的硬件平台。

据悉,最新 MLPerf Training v3.0 包含了基于 GPT-3 175B 的大型语言模型(LLM)测试,侧重于生成式 AI 能力。

图片

图源:MLPerf benchmarks。

LLM 训练测试中还使用了专注于 GPU 加速的云计算服务提供商 CoreWeave 提供的 NVIDIA HGX H100 基础设施,在多个规模上联合提交了 LLM 工作负载。

出来的结果令人振奋:在 896 个英特尔 Xeon Platinum 8462Y 和 3584 个英伟达 H100 芯片的合作下,仅仅用了 11 分钟就完成了基于 GPT-3 的 LLM 训练任务。

图片图片

一时之间,AI 社区响起了老黄赢麻了的欢呼声。

可结果真如此吗?今天有人对此事提出了质疑。

先是 PyTorch 创始人 Soumith Chintala,他认为 GPT-3 并没有在 11 分钟内训练完成。使用 3584 个 H100 GPU,GPT-3 架构在 C4 数据集上训练了 11 分钟,对数概率为 2.69。

这里不要只关注「11 分钟」,因为这就像说「ResNet-50 在 MNIST(其实这里他想说的是 CIFAR100)上,5 秒内训练达到了 80%的准确率。」

图片图片

推特用户 @abhi_venigalla 对 Soumith 的观点表示赞同,他是生成式 AI 创业公司 MosaicML 的研究者。他认为 MLPerf 基准测试中训练 GPT-3 的时间更可能是 2 天。

在他看来,该基准测试并不是完整的 GPT-3。MLPerf 的 LLM 基准只提供了一个开始的检查点,然后必须达到目标损失。

因此,该基准 11 分钟跑完 GPT-3 只覆盖 1.2B 的 token,而非完整的 300B token。此外看起来更像是 540 TFLOPs/H100,从配置上看像 FP8,~27% MFU。

不过他也承认,从 H100 在其软件生命周期早期这一点来看,以此规模实现现在这种性能也相当惊人。

图片图片

所以,最新 MLPerf 基准测试中 11 分钟训完 GPT-3 是不是被误解呢?评论区的小伙伴也可以发表下自己的观点。

责任编辑:张燕妮 来源: 机器之心
相关推荐

2023-06-29 21:30:30

英伟达用户基础设施

2023-06-29 17:40:10

模型测试

2023-11-10 15:35:52

AI模型

2023-07-13 23:16:19

英伟达GPU

2023-08-06 13:01:34

AI开发

2023-11-21 09:14:33

微软Azure AI

2023-06-02 12:10:43

OpenAI版本GPT

2009-06-19 16:44:56

SpringJava EE

2023-09-14 13:23:00

AI芯片

2020-08-07 15:43:03

Python表格代码

2020-06-16 08:47:53

磁盘

2024-03-14 14:49:34

Meta人工智能

2024-04-10 09:10:27

Gaudi 3芯片英特尔

2023-05-11 08:48:58

谷歌超级计算机

2021-02-20 08:30:40

Netflix996文化

2021-03-03 15:34:57

人工智能机器学习技术

2023-08-24 14:26:00

数据中心利润AI

2023-03-28 09:38:10

2018-03-27 11:37:02

iOS程序面试

2013-03-05 13:23:29

Linus TorvaGNOME 3
点赞
收藏

51CTO技术栈公众号