英伟达下场,首次优化DeepSeek-R1!B200性能狂飙25倍,碾压H100

当 FP4 的魔法与 Blackwell 的强大算力相遇,会碰撞出怎样的火花?

答案是:推理性能暴涨 25 倍,成本狂降 20 倍!

随着 DeepSeek-R1 本地化部署的爆火,英伟达也亲自下场,开源了首个基于 Blackwell 架构的优化方案------DeepSeek-R1-FP4。

在新模型的加持下,B200 实现了高达 21,088 token 每秒的的推理吞吐量,相比于 H100 的 844 token 每秒,提升了 25 倍。

与此同时,每 token 的成本也实现了 20 倍的降低。

通过在 Blackwell 架构上应用 TensorRT DeepSeek 优化,英伟达让具有 FP4 生产级精度的模型,在 MMLU 通用智能基准测试中达到了 FP8 模型性能的 99.8%。

DeepSeek-R1 首次基于 Blackwell GPU 优化

目前,英伟达基于 FP4 优化的 DeepSeek-R1 检查点现已在 Hugging Face 上开源。

模型地址:huggingface.co/nvidia/Deep...

后训练量化

模型将 Transformer 模块内的线性算子的权重和激活量化到了 FP4,适用于 TensorRT-LLM 推理。

这种优化将每个参数从 8 位减少到 4 位,从而让磁盘空间和 GPU 显存的需求减少了约 1.6 倍。

使用 TensorRT-LLM 部署

要使用 TensorRT-LLM LLM API 部署量化后的 FP4 权重文件,并为给定的提示生成文本响应,请参照以下示例代码:

硬件要求:需要支持 TensorRT-LLM 的英伟达 GPU(如 B200),并且需要 8 个 GPU 来实现 tensor_parallel_size=8 的张量并行。

性能优化:代码利用 FP4 量化、TensorRT 引擎和并行计算,旨在实现高效、低成本的推理,适合生产环境或高吞吐量应用。

对于此次优化的成果,网友表示惊叹。

「FP4 魔法让 AI 未来依然敏锐!」网友 Isha 评论道。

网友 algorusty 则声称,有了这次的优化后,美国供应商能够以每百万 token 0.25 美元的价格提供 R1。

「还会有利润。」

网友 Phil 则将这次的优化与 DeepSeek 本周的开源 5 连发结合了起来。

「这展示了硬件和开源模型结合的可能性。」他表示。

DeepSeek 全面开源

如今 DeepSeek 持续 5 天的「开源周」已经进行到了第 3 天。

周一,他们开源了 FlashMLA。这是 DeepSeek 专为英伟达 Hopper GPU 打造的高效 MLA 解码内核,特别针对变长序列进行了优化,目前已正式投产使用。

周二开源了 DeepEP,这是一个专为混合专家系统(MoE)和专家并行(EP)设计的通信库。

周三开源的是 DeepGEMM。这是一个支持稠密和 MoE 模型的 FP8 GEMM(通用矩阵乘法)计算库,可为 V3/R1 的训练和推理提供强大支持。

总的来说,不管是英伟达开源的 DeepSeek-R1-FP4,还是 DeepSeek 开源的三个仓库,都是通过对英伟达 GPU 和集群的优化,来推动 AI 模型的高效计算和部署。

参考资料:

x.com/NVIDIAAIDev...

相关推荐
寻道码路11 分钟前
深度剖析 Video-RAG:厦门大学和罗切斯特大学联合推出的一种用于长视频理解的检索增强生成技术
人工智能·语言模型·开源·aigc·音视频·ai编程
一只蜗牛儿14 分钟前
Sherpa-ONNX:说话人识别与语音识别自动开启(VAD)+ Python API 完整指南
人工智能·python·语音识别
结衣结衣.14 分钟前
【OpenCV】入门教学
图像处理·人工智能·python·opencv
cly021631 分钟前
12个大语言模型平台对比测试-搜索视角
人工智能·语言模型·自然语言处理
lyyiangang43 分钟前
记一次pytorch训练loss异常的问题
人工智能·pytorch·python
Y_3_71 小时前
30 分钟从零开始入门 CSS
开发语言·前端·css·人工智能·python·tensorflow
r_martian1 小时前
人工智能(AI):科技新纪元的领航者
人工智能·科技
2403_875180951 小时前
AI数字人开发,引领科技新潮流
java·大数据·数据结构·人工智能·科技·前端框架
黑金IT1 小时前
打造优质 DeepSeek “相关链接” 功能:提升用户体验的深度剖析
人工智能·知识图谱·deepseek
百事不可口y1 小时前
【产品小白】怎么量化用户体验呢
大数据·人工智能·产品运营·产品经理·用户运营·内容运营