面向 MoE 专家并行的 GPU 通信库,提供 dispatch 与 combine 所需的高吞吐、低延迟 all-to-all 内核。
deepseek-ai/DeepInfer
DeepSeek 开源的离线推理框架,用于在 NVIDIA SM100 和华为昇腾 950 硬件上运行 DeepSeek-V4.1-Flash。
- Star
- 8,284
- Fork
- 697
- 未关闭 Issue
- 78
- 最近提交
- 昨天
这是什么?
DeepInfer 是 DeepSeek 推出的推理框架,面向大规模部署 DeepSeek 大模型。当前开源版本支持 DeepSeek-V4.1-Flash 的离线推理,包含优化的算子内核、模型与采样器运行时,并支持注意力数据并行、上下文并行,以及带 EPLB 的 MoE 专家并行。
框架提供基于 DSpark 的投机解码与置信度调度验证,通过有界 SWA 回放减少解码器预填充计算,并采用异步调度器让 CPU 批调度与设备执行重叠。此外还支持批次不变推理、确定性采样,以及按正则、JSON Schema 或 EBNF 文法约束输出的结构化生成。
运行环境为 Linux 与 Python 3.12,NVIDIA 平台需要 CUDA 13,昇腾平台需要 CANN 9.2。该版本不包含 KV 缓存系统以及 Prefill/Decode 分离部署,这些是构建完整生产服务所需的部分。
核心功能
- 01针对 DeepSeek-V4.1-Flash 的离线推理,包含优化算子内核与采样器运行时
- 02支持注意力数据并行、上下文并行,以及带 EPLB 的 MoE 专家并行
- 03基于 DSpark 的投机解码,并采用置信度调度验证
- 04异步批调度,使 CPU 工作与设备执行重叠进行
- 05在 NVIDIA GPU 上支持批次不变推理与确定性采样
- 06支持按正则、JSON Schema 或 EBNF 文法约束的结构化输出
适用场景
- 在 NVIDIA SM100 GPU 上使用 DeepSeek-V4.1-Flash 进行离线批量推理
- 让大模型生成符合 JSON Schema 等约束的结构化输出
- 研究 DeepSeek 如何通过专家并行与投机解码部署 MoE 模型
- 在华为昇腾 950 NPU 上部署 DeepSeek 模型
快速上手
- 1
创建并激活一个空的 Python 3.12 环境
- 2
将加速器设置为 NVIDIA
export DEEPINFER_ACCELERATOR=nvidia - 3
从源码构建 DeepInfer
./build.sh - 4
运行离线推理(首次运行时会下载默认模型权重)
python examples/offline_inference.py
使用前须知
- 仅支持 NVIDIA SM100 GPU 或华为 Ascend 950 NPU,需要 Linux 系统,并依赖 CUDA 13 或 CANN 9.2。
- 仅支持 DeepSeek-V4.1-Flash 模型。
- 此开源版本不包含 KV 缓存系统以及预填充/解码分离,因此不是完整的推理服务栈。
- 部分功能(如批次不变推理和确定性采样)仅适用于 NVIDIA。
标签
常见问题
- DeepInfer 是什么?
- DeepSeek 开源的离线推理框架,用于在 NVIDIA SM100 和华为昇腾 950 硬件上运行 DeepSeek-V4.1-Flash。 DeepInfer 是 DeepSeek 推出的推理框架,面向大规模部署 DeepSeek 大模型。当前开源版本支持 DeepSeek-V4.1-Flash 的离线推理,包含优化的算子内核、模型与采样器运行时,并支持注意力数据并行、上下文并行,以及带 EPLB 的 MoE 专家并行。
- DeepInfer 可以免费使用吗?用的什么许可证?
- DeepInfer 以 Apache-2.0 许可证开源。用于商业产品前,请先确认许可证条款。
- DeepInfer 还在维护吗?
- 活跃维护中。DeepInfer 最近一次提交是在 2026年10月20日。
- DeepInfer 适合用来做什么?
- 在 NVIDIA SM100 GPU 上使用 DeepSeek-V4.1-Flash 进行离线批量推理 让大模型生成符合 JSON Schema 等约束的结构化输出 研究 DeepSeek 如何通过专家并行与投机解码部署 MoE 模型 在华为昇腾 950 NPU 上部署 DeepSeek 模型
- DeepInfer 怎么安装和运行?
- 创建并激活一个空的 Python 3.12 环境 将加速器设置为 NVIDIA: export DEEPINFER_ACCELERATOR=nvidia 从源码构建 DeepInfer: ./build.sh 运行离线推理(首次运行时会下载默认模型权重): python examples/offline_inference.py
- 在哪里获取 DeepInfer?
- 源代码和安装说明在 GitHub 上:https://github.com/deepseek-ai/DeepInfer
概要由 AI 根据项目 README 生成;GitHub 数据每日更新。
相关项目
一个高吞吐、低显存占用的大语言模型推理与服务引擎,提供兼容 OpenAI 的 API。
SGLang 是一个开源推理框架,用于在多种 GPU、TPU、CPU 等硬件上部署大语言模型、视觉语言模型和扩散模型。
面向消费级硬件的原生推理引擎,用于在 Metal、CUDA 和 ROCm 上运行 DeepSeek V4 Flash 等少数大模型。
用于压缩、部署和服务大语言模型与视觉语言模型的工具包,内含 TurboMind 与 PyTorch 两种推理引擎。
EAGLE-4 的推理代码,通过投机解码为 SGLang、vLLM 和 TokenSpeed 上的大模型推理提速。