DeepSeek-V4-Pro¶
1 引言¶
DeepSeek-V4 在 DeepSeek-V3 的基础上引入了若干关键升级:
- 流形约束超连接(Manifold-Constrained Hyper-Connections,mHC),用于增强传统的残差连接。
- 一种混合注意力架构,通过 Compress-4-Attention 和 Compress-128-Attention 大幅提升长上下文效率。对于混合专家(Mixture-of-Experts,MoE)组件,仍然采用 DeepSeekMoE 架构,仅做了少量调整。
DeepSeek-V4-Pro 是 DeepSeek-V4 的最大推理努力模式,显著提升了开源模型的知识能力,稳固确立了其作为当今最佳开源模型的地位。它在编码基准测试中取得顶级性能,并在推理和智能体任务上大幅缩小了与领先闭源模型的差距。
本文档将介绍该模型的主要验证步骤,包括支持的特性、特性配置、环境准备、单节点与多节点部署、精度和性能评估。
2 支持的特性¶
请参阅 支持的特性列表 获取模型支持的特性矩阵。
请参阅 特性指南 获取特性配置。
3 前提条件¶
3.1 模型权重¶
| 权重版本 | 硬件要求 | 下载链接 |
|---|---|---|
DeepSeek-V4-Pro-w4a8-mtp |
2 个 Atlas 800 A3(128GB × 8)节点或 4 个 Atlas 800 A2(64GB × 8)节点 | ModelScope |
DeepSeek-V4-Pro-0813-w4a8 |
ModelScope |
DeepSeek-V4-Pro-0813-w4a8(量化后集成 DSpark 的官方发布版):该检查点包含 DSpark 草稿权重,因此无需单独的草稿模型路径。
建议将模型权重下载到多节点共享目录,例如 /root/.cache/。
路径说明:将模型权重下载到你选择的目录并记录下来。确保后续部署命令中的模型路径与该目录一致。
3.2 验证多节点通信(可选)¶
如果要部署多节点环境,需要根据 验证多节点通信环境 验证多节点通信。
4 安装¶
4.1 Docker 镜像安装¶
根据你的机器类型选择镜像,并在节点上启动 Docker 镜像,请参阅 使用 Docker。
在每个节点上启动 Docker 镜像。
export IMAGE=quay.io/ascend/vllm-ascend:v0.23.0-a3
docker run --rm \
--name vllm-ascend \
--shm-size=512g \
--net=host \
--privileged=true \
--device /dev/davinci0 \
--device /dev/davinci1 \
--device /dev/davinci2 \
--device /dev/davinci3 \
--device /dev/davinci4 \
--device /dev/davinci5 \
--device /dev/davinci6 \
--device /dev/davinci7 \
--device /dev/davinci8 \
--device /dev/davinci9 \
--device /dev/davinci10 \
--device /dev/davinci11 \
--device /dev/davinci12 \
--device /dev/davinci13 \
--device /dev/davinci14 \
--device /dev/davinci15 \
--device /dev/davinci_manager \
--device /dev/devmm_svm \
--device /dev/hisi_hdc \
-v /usr/local/dcmi:/usr/local/dcmi \
-v /usr/local/Ascend/driver/tools/hccn_tool:/usr/local/Ascend/driver/tools/hccn_tool \
-v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi \
-v /usr/local/Ascend/driver/lib64/:/usr/local/Ascend/driver/lib64/ \
-v /usr/local/Ascend/driver/version.info:/usr/local/Ascend/driver/version.info \
-v /etc/ascend_install.info:/etc/ascend_install.info \
-v /etc/hccn.conf:/etc/hccn.conf \
-v /root/.cache:/root/.cache \
-it $IMAGE bash
在每个节点上启动 Docker 镜像。
export IMAGE=quay.io/ascend/vllm-ascend:v0.23.0
docker run --rm \
--name vllm-ascend \
--shm-size=512g \
--net=host \
--privileged=true \
--device /dev/davinci0 \
--device /dev/davinci1 \
--device /dev/davinci2 \
--device /dev/davinci3 \
--device /dev/davinci4 \
--device /dev/davinci5 \
--device /dev/davinci6 \
--device /dev/davinci7 \
--device /dev/davinci_manager \
--device /dev/devmm_svm \
--device /dev/hisi_hdc \
-v /usr/local/dcmi:/usr/local/dcmi \
-v /usr/local/Ascend/driver/tools/hccn_tool:/usr/local/Ascend/driver/tools/hccn_tool \
-v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi \
-v /usr/local/Ascend/driver/lib64/:/usr/local/Ascend/driver/lib64/ \
-v /usr/local/Ascend/driver/version.info:/usr/local/Ascend/driver/version.info \
-v /etc/ascend_install.info:/etc/ascend_install.info \
-v /etc/hccn.conf:/etc/hccn.conf \
-v /root/.cache:/root/.cache \
-it $IMAGE bash
Docker 运行成功后,可以通过执行 docker ps 命令验证正在运行的容器服务。
4.2 源码安装¶
如果你不想像上面那样使用 Docker 镜像,也可以从源码构建所有内容:
- 从源码安装
vllm-ascend,请参阅 安装。
如果要部署多节点环境,需要在每个节点上配置环境。
5 在线服务部署¶
Note
在本教程中,我们假设您已将模型权重下载到 /root/.cache/modelscope/hub/models/vllm-ascend/。您可以将其更改为自己的路径。
建议将以下服务代码封装在 .sh 脚本文件中,并在 Bash 模式下执行。
5.1 多节点在线部署¶
量化模型 DeepSeek-V4-Pro-w4a8-mtp 至少需要 2 个 Atlas 800 A3(128GB × 8)节点或 4 个 Atlas 800 A2(64GB × 8)节点。请分别在每个节点上运行以下脚本。
节点0
local_ip="xxx"
node0_ip="xxxx"
export HCCL_IF_IP=$local_ip
export IFNAME="xxx"
export GLOO_SOCKET_IFNAME="$IFNAME"
export TP_SOCKET_IFNAME="$IFNAME"
export HCCL_SOCKET_IFNAME="$IFNAME"
export HCCL_BUFFSIZE=512
export ASCEND_RT_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
export OMP_PROC_BIND=false
export OMP_NUM_THREADS=10
export PYTORCH_NPU_ALLOC_CONF=expandable_segments:True
export ACL_OP_INIT_MODE=1
export VLLM_ENGINE_READY_TIMEOUT_S=3600
export HCCL_OP_EXPANSION_MODE="AIV"
export TASK_QUEUE_ENABLE=1
export LD_PRELOAD=/usr/lib/aarch64-linux-gnu/libjemalloc.so.2:$LD_PRELOAD
export HCCL_CONNECT_TIMEOUT=7200
export ASCEND_CONNECT_TIMEOUT=10000
export ASCEND_TRANSFER_TIMEOUT=10000
export VLLM_RPC_TIMEOUT=1800000
# Ensure the model path matches the directory recorded during download
vllm serve /root/.cache/modelscope/hub/models/vllm-ascend/DeepSeek-V4-Pro-w4a8-mtp \
--host 0.0.0.0 \
--port 8000 \
--data-parallel-address $node0_ip \
--data-parallel-size 4 \
--data-parallel-size-local 1 \
--data-parallel-start-rank 0 \
--tensor-parallel-size 8 \
--enable-expert-parallel \
--served-model-name dsv4 \
--max-model-len 135000 \
--max-num-batched-tokens 4096 \
--max-num-seqs 16 \
--gpu-memory-utilization 0.9 \
--block-size 128 \
--no-enable-prefix-caching \
--tokenizer-mode deepseek_v4 \
--tool-call-parser deepseek_v4 \
--enable-auto-tool-choice \
--reasoning-parser deepseek_v4 \
--attention_config.indexer_kv_dtype int8 \
--model-loader-extra-config '{
"enable_multithread_load": true,
"num_threads": 128
}' \
--quantization ascend \
--speculative-config '{
"num_speculative_tokens": 1,
"method": "mtp",
"enforce_eager": true
}' \
--compilation-config '{
"cudagraph_mode":"FULL_DECODE_ONLY"
}' \
--additional-config '{
"ascend_compilation_config":{
"enable_npugraph_ex":true,
"enable_static_kernel":false
},
"enable_cpu_binding": true,
"enable_shared_expert_dp": true,
"enable_flashcomm1": true,
"multistream_overlap_shared_expert":true
}'
节点1-节点3
local_ip="xxx"
node0_ip="xxxx"
export HCCL_IF_IP=$local_ip
export IFNAME="xxx"
export GLOO_SOCKET_IFNAME="$IFNAME"
export TP_SOCKET_IFNAME="$IFNAME"
export HCCL_SOCKET_IFNAME="$IFNAME"
export HCCL_BUFFSIZE=512
export ASCEND_RT_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
export OMP_PROC_BIND=false
export OMP_NUM_THREADS=10
export PYTORCH_NPU_ALLOC_CONF=expandable_segments:True
export ACL_OP_INIT_MODE=1
export VLLM_ENGINE_READY_TIMEOUT_S=3600
export HCCL_OP_EXPANSION_MODE="AIV"
export TASK_QUEUE_ENABLE=1
export LD_PRELOAD=/usr/lib/aarch64-linux-gnu/libjemalloc.so.2:$LD_PRELOAD
export HCCL_CONNECT_TIMEOUT=7200
export ASCEND_CONNECT_TIMEOUT=10000
export ASCEND_TRANSFER_TIMEOUT=10000
export VLLM_RPC_TIMEOUT=1800000
# Ensure the model path matches the directory recorded during download
vllm serve /root/.cache/modelscope/hub/models/vllm-ascend/DeepSeek-V4-Pro-w4a8-mtp \
--host 0.0.0.0 \
--port 8000 \
--headless \
--data-parallel-address $node0_ip \
--data-parallel-size 4 \
--data-parallel-size-local 1 \
--data-parallel-start-rank 1 \
--tensor-parallel-size 8 \
--enable-expert-parallel \
--served-model-name dsv4 \
--max-model-len 135000 \
--max-num-batched-tokens 4096 \
--max-num-seqs 16 \
--gpu-memory-utilization 0.9 \
--block-size 128 \
--no-enable-prefix-caching \
--tokenizer-mode deepseek_v4 \
--tool-call-parser deepseek_v4 \
--enable-auto-tool-choice \
--reasoning-parser deepseek_v4 \
--attention_config.indexer_kv_dtype int8 \
--model-loader-extra-config '{
"enable_multithread_load": true,
"num_threads": 128
}' \
--quantization ascend \
--speculative-config '{
"num_speculative_tokens": 1,
"method": "mtp",
"enforce_eager": true
}' \
--compilation-config '{
"cudagraph_mode":"FULL_DECODE_ONLY"
}' \
--additional-config '{
"ascend_compilation_config":{
"enable_npugraph_ex":true,
"enable_static_kernel":false
},
"enable_cpu_binding": true,
"enable_shared_expert_dp": true,
"enable_flashcomm1": true,
"multistream_overlap_shared_expert":true
}'
节点0
# this obtained through ifconfig
# nic_name is the network interface name corresponding to local_ip of the current node
nic_name="xxx"
local_ip="xxx"
# The value of node0_ip must be consistent with the value of local_ip set in node0 (master node)
node0_ip="xxxx"
export HCCL_OP_EXPANSION_MODE="AIV"
export HCCL_IF_IP=$local_ip
export GLOO_SOCKET_IFNAME=$nic_name
export TP_SOCKET_IFNAME=$nic_name
export HCCL_SOCKET_IFNAME=$nic_name
export HCCL_BUFFSIZE=2048
export PYTORCH_NPU_ALLOC_CONF=expandable_segments:True
export OMP_PROC_BIND=false
export OMP_NUM_THREADS=10
export TASK_QUEUE_ENABLE=1
export LD_PRELOAD=/usr/lib/aarch64-linux-gnu/libjemalloc.so.2:$LD_PRELOAD
# Ensure the model path matches the directory recorded during download
vllm serve /root/.cache/modelscope/hub/models/vllm-ascend/DeepSeek-V4-Pro-w4a8-mtp \
--host 0.0.0.0 \
--port 8000 \
--data-parallel-address $node0_ip \
--data-parallel-rpc-port 13399 \
--data-parallel-size 2 \
--data-parallel-size-local 1 \
--data-parallel-start-rank 0 \
--tensor-parallel-size 16 \
--enable-expert-parallel \
--served-model-name dsv4 \
--max-model-len 135000 \
--max-num-batched-tokens 4096 \
--max-num-seqs 32 \
--gpu-memory-utilization 0.9 \
--block-size 128 \
--tokenizer-mode deepseek_v4 \
--tool-call-parser deepseek_v4 \
--enable-auto-tool-choice \
--reasoning-parser deepseek_v4 \
--attention_config.indexer_kv_dtype int8 \
--model-loader-extra-config='{"enable_multithread_load": true, "num_threads": 128}' \
--quantization ascend \
--speculative-config '{"num_speculative_tokens": 1,"method": "mtp","enforce_eager": true}' \
--compilation-config '{"cudagraph_mode":"FULL_DECODE_ONLY"}' \
--additional-config '
{"ascend_compilation_config":{
"enable_npugraph_ex":true,
"enable_static_kernel":false
},
"enable_cpu_binding": true,
"enable_flashcomm1": true,
"multistream_overlap_shared_expert":true}'
节点1
```bash
# this obtained through ifconfig
# nic_name is the network interface name corresponding to local_ip of the current node
nic_name="xxx"
local_ip="xxx"
# The value of node0_ip must be consistent with the value of local_ip set in node0 (master node)
node0_ip="xxxx"
export HCCL_OP_EXPANSION_MODE="AIV"
export HCCL_IF_IP=$local_ip
export GLOO_SOCKET_IFNAME=$nic_name
export TP_SOCKET_IFNAME=$nic_name
export HCCL_SOCKET_IFNAME=$nic_name
export HCCL_BUFFSIZE=2048
export PYTORCH_NPU_ALLOC_CONF=expandable_segments:True
export OMP_PROC_BIND=false
export OMP_NUM_THREADS=10
export TASK_QUEUE_ENABLE=1
export LD_PRELOAD=/usr/lib/aarch64-linux-gnu/libjemalloc.so.2:$LD_PRELOAD
# Ensure the model path matches the directory recorded during download
vllm serve /root/.cache/modelscope/hub/models/vllm-ascend/DeepSeek-V4-Pro-w4a8-mtp \
--host 0.0.0.0 \
--port 8000 \
--headless \
--data-parallel-address $node0_ip \
--data-parallel-rpc-port 13399 \
--data-parallel-size 2 \
--data-parallel-size-local 1 \
--data-parallel-start-rank 1 \
--tensor-parallel-size 16 \
--enable-expert-parallel \
--served-model-name dsv4 \
--max-model-len 135000 \
--max-num-batched-tokens 4096 \
--max-num-seqs 32 \
--gpu-memory-utilization 0.9 \
--block-size 128 \
--tokenizer-mode deepseek_v4 \
--tool-call-parser deepseek_v4 \
--enable-auto-tool-choice \
--reasoning-parser deepseek_v4 \
--attention_config.indexer_kv_dtype int8 \
--model-loader-extra-config='{"enable_multithread_load": true, "num_threads": 128}' \
--quantization ascend \
--speculative-config '{"num_speculative_tokens": 1,"method": "mtp","enforce_eager": true}' \
--compilation-config '{"cudagraph_mode":"FULL_DECODE_ONLY"}' \
--additional-config '
{"ascend_compilation_config":{
"enable_npugraph_ex":true,
"enable_static_kernel":false
},
"enable_cpu_binding": true,
"enable_flashcomm1": true,
"multistream_overlap_shared_expert":true}'
```
在两个节点上使用官方 DeepSeek-V4-Pro-0813-w4a8 检查点。该示例保留现有的 DP2/TP16 拓扑,仅更改 DSpark 相关和必需的运行时参数。
节点0
# nic_name is the network interface that owns local_ip on the current node.
nic_name="xxx"
local_ip="xxx"
# node0_ip must be the local_ip of Node0 on every node.
node0_ip="xxxx"
export HCCL_IF_IP=$local_ip
export GLOO_SOCKET_IFNAME=$nic_name
export TP_SOCKET_IFNAME=$nic_name
export HCCL_SOCKET_IFNAME=$nic_name
export HCCL_OP_EXPANSION_MODE="AIV"
export HCCL_BUFFSIZE=1024
export HCCL_CONNECT_TIMEOUT=7200
export ASCEND_CONNECT_TIMEOUT=10000
export ASCEND_TRANSFER_TIMEOUT=10000
export VLLM_RPC_TIMEOUT=1800000
export PYTORCH_NPU_ALLOC_CONF=expandable_segments:True
export OMP_PROC_BIND=false
export OMP_NUM_THREADS=10
export TASK_QUEUE_ENABLE=1
export LD_PRELOAD=/usr/lib/aarch64-linux-gnu/libjemalloc.so.2:$LD_PRELOAD
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib/
export VLLM_PREFIX_CACHE_RETENTION_INTERVAL=4096
vllm serve /path/to/DeepSeek-V4-Pro-0813-w4a8 \
--host 0.0.0.0 \
--port 8000 \
--data-parallel-address $node0_ip \
--data-parallel-rpc-port 13399 \
--data-parallel-size 2 \
--data-parallel-size-local 1 \
--data-parallel-start-rank 0 \
--tensor-parallel-size 16 \
--enable-expert-parallel \
--served-model-name dsv4 \
--max-model-len 135000 \
--max-num-batched-tokens 4096 \
--max-num-seqs 16 \
--gpu-memory-utilization 0.9 \
--block-size 32 \
--tokenizer-mode deepseek_v4 \
--tool-call-parser deepseek_v4 \
--enable-auto-tool-choice \
--reasoning-parser deepseek_v4 \
--attention_config.indexer_kv_dtype int8 \
--model-loader-extra-config='{"enable_multithread_load": true, "num_threads": 128}' \
--quantization ascend \
--speculative-config '{"num_speculative_tokens":5,"method":"dspark","enforce_eager":true}' \
--compilation-config '{"cudagraph_mode":"FULL_DECODE_ONLY"}' \
--additional-config '{
"ascend_compilation_config": {
"enable_npugraph_ex": true,
"enable_static_kernel": false
},
"enable_cpu_binding": true,
"enable_flashcomm1": true,
"multistream_overlap_shared_expert": true
}'
节点1
使用与节点0相同的环境变量和命令,并进行以下 DP 更改:
nic_name="xxx"
local_ip="xxx"
node0_ip="xxxx"
export HCCL_IF_IP=$local_ip
export GLOO_SOCKET_IFNAME=$nic_name
export TP_SOCKET_IFNAME=$nic_name
export HCCL_SOCKET_IFNAME=$nic_name
export HCCL_OP_EXPANSION_MODE="AIV"
export HCCL_BUFFSIZE=1024
export HCCL_CONNECT_TIMEOUT=7200
export ASCEND_CONNECT_TIMEOUT=10000
export ASCEND_TRANSFER_TIMEOUT=10000
export VLLM_RPC_TIMEOUT=1800000
export PYTORCH_NPU_ALLOC_CONF=expandable_segments:True
export OMP_PROC_BIND=false
export OMP_NUM_THREADS=10
export TASK_QUEUE_ENABLE=1
export LD_PRELOAD=/usr/lib/aarch64-linux-gnu/libjemalloc.so.2:$LD_PRELOAD
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib/
export VLLM_PREFIX_CACHE_RETENTION_INTERVAL=4096
vllm serve /path/to/DeepSeek-V4-Pro-0813-w4a8 \
--host 0.0.0.0 \
--port 8000 \
--headless \
--data-parallel-address $node0_ip \
--data-parallel-rpc-port 13399 \
--data-parallel-size 2 \
--data-parallel-size-local 1 \
--data-parallel-start-rank 1 \
--tensor-parallel-size 16 \
--enable-expert-parallel \
--served-model-name dsv4 \
--max-model-len 135000 \
--max-num-batched-tokens 4096 \
--max-num-seqs 16 \
--gpu-memory-utilization 0.9 \
--block-size 32 \
--tokenizer-mode deepseek_v4 \
--tool-call-parser deepseek_v4 \
--enable-auto-tool-choice \
--reasoning-parser deepseek_v4 \
--attention_config.indexer_kv_dtype int8 \
--model-loader-extra-config='{"enable_multithread_load": true, "num_threads": 128}' \
--quantization ascend \
--speculative-config '{"num_speculative_tokens":5,"method":"dspark","enforce_eager":true}' \
--compilation-config '{"cudagraph_mode":"FULL_DECODE_ONLY"}' \
--additional-config '{
"ascend_compilation_config": {
"enable_npugraph_ex": true,
"enable_static_kernel": false
},
"enable_cpu_binding": true,
"enable_flashcomm1": true,
"multistream_overlap_shared_expert": true
}'
关键参数说明:
--data-parallel-size设置全局数据并行 rank 数量,--data-parallel-size-local设置当前节点上的 DP rank 数量。--data-parallel-start-rank指定当前节点的数据并行起始 rank。每个节点必须设置为唯一值(例如,Node0 = 0,Node1 = 1)。--data-parallel-address指定数据并行主节点(Node0)的 IP 地址。它必须在所有节点上保持一致。--data-parallel-rpc-port是 DP RPC 端口。所有节点使用相同值,并确保端口可用。--tensor-parallel-size设置每个 DP rank 内的张量并行大小。根据部署拓扑和可用 NPU 数量,与 DP 大小一起配置。--enable-expert-parallel为 MoE 层启用专家并行。不要在同一 MoE 层中混用 MoE 张量并行和专家并行。--headless(用于非主节点)禁用该节点上的 API 服务器,因为只有主节点处理请求。--max-model-len指定最大上下文长度。根据实际场景调整。--max-num-seqs表示每个 DP 组允许处理的最大请求数。如果发送到服务的请求数超过此限制,超出请求将保持等待状态且不会被调度。注意,等待状态所花费的时间也会计入 TTFT 和 TPOT 等指标。因此,性能测试时通常建议--max-num-seqs*--data-parallel-size>= 实际总并发数。--max-num-batched-tokens是单个调度步骤中处理的最大 token 数。更大的值可提高 prefill 效率,但会消耗更多激活内存。--tokenizer-mode deepseek_v4、--tool-call-parser deepseek_v4、--enable-auto-tool-choice和--reasoning-parser deepseek_v4启用 DeepSeek-V4 tokenizer 行为、自动工具调用和推理输出解析。--no-enable-prefix-caching表示禁用前缀缓存。若要启用,请移除此选项。--block-size设置 KV 缓存块大小。若要启用实验性 4k 前缀缓存命中支持,请将其从128改为32。--quantization ascend为 W4A8 模型启用 Ascend 量化。--model-loader-extra-config='{"enable_multithread_load": true, "num_threads": 128}'选择多线程权重迭代器。enable_multithread_load必须是 JSON 布尔值,num_threads必须是正整数。--safetensors-load-strategy prefetch是一种替代方案,在常规迭代器加载之前将检查点文件预热到操作系统页缓存中。不要与多线程加载组合使用:当enable_multithread_load为true时,默认加载器会拒绝prefetch、eager或torchao。本文档中的示例仅使用多线程加载。--speculative-config配置推测解码。预览 MTP 检查点使用mtp,DeepSeek-V4-Pro-0813-w4a8使用dspark。对于 DSpark,使用检查点声明的值;示例使用五个推测 token,所有 rank 必须使用相同值。--compilation-config '{"cudagraph_mode":"FULL_DECODE_ONLY"}'在解码阶段启用完整 ACL 图执行,以降低调度延迟。--additional-config启用 Ascend 特定优化。enable_npugraph_ex启用增强型 ACL 图执行,enable_static_kernel: false保持禁用静态内核编译,enable_cpu_binding启用 Ascend 原生 CPU 绑定,enable_shared_expert_dp为共享专家启用数据并行,multistream_overlap_shared_expert重叠共享专家计算以提高 MoE 吞吐量。--additional-config中的enable_flashcomm1: true启用 FlashComm1 通信优化。这是已弃用的VLLM_ASCEND_ENABLE_FLASHCOMM1环境变量的推荐替代方案。只要启用enable_dsa_cp,就会自动启用。VLLM_PREFIX_CACHE_RETENTION_INTERVAL=4096每 4096 个 token 保留前缀缓存检查点。它仅在前缀缓存启用时生效,且必须是--block-size的非负倍数;4096与 DSpark 示例的块大小32匹配。
常见问题提示:如果遇到问题,请参阅公共常见问题解答进行故障排除。
服务验证:
curl http://<node0_ip>:8000/v1/chat/completions \
-H "Content-Type: application/json" \
-d '{
"model": "dsv4",
"messages": [
{
"role": "user",
"content": "Who are you?"
}
],
"max_tokens": 256,
"temperature": 0
}'
预期结果:
服务返回 HTTP 200 OK,JSON 响应中包含 choices 字段。
5.2 多节点 PD 分离部署¶
我们建议使用 Mooncake 进行部署:Mooncake。
在标准部署模式下,Prefill(prompt 处理)和 Decode(token 生成)任务运行在同一组 NPU 上。PD(Prefill-Decode)分离通过将 Prefill 和 Decode 运行在专用节点组上来解决此问题,每个节点组独立配置。对于需要稳定延迟和高吞吐量的并发多用户工作负载的生产部署,推荐采用此架构。
以下章节描述在 Atlas 800 A3(128GB × 8)和 Atlas 800 A2(64GB × 8)多节点环境上的 PD 分离部署。
5.2.1 A3 系列 PD 分离部署¶
本节展示在 Atlas 800 A3(128GB × 8)上部署一个 Prefill 池和一个 Decode 池的示例。示例中每个池使用两个物理节点。
开始之前,请:
-
在每个节点上准备脚本
launch_online_dp.py。import argparse import multiprocessing import os import subprocess import sys def parse_args(): parser = argparse.ArgumentParser() parser.add_argument( "--dp-size", type=int, required=True, help="Data parallel size." ) parser.add_argument( "--tp-size", type=int, default=1, help="Tensor parallel size." ) parser.add_argument( "--dp-size-local", type=int, default=-1, help="Local data parallel size." ) parser.add_argument( "--dp-rank-start", type=int, default=0, help="Starting rank for data parallel." ) parser.add_argument( "--dp-address", type=str, required=True, help="IP address for data parallel master node." ) parser.add_argument( "--dp-rpc-port", type=str, default=12345, help="Port for data parallel master node." ) parser.add_argument( "--vllm-start-port", type=int, default=9000, help="Starting port for the engine." ) return parser.parse_args() args = parse_args() dp_size = args.dp_size tp_size = args.tp_size dp_size_local = args.dp_size_local if dp_size_local == -1: dp_size_local = dp_size dp_rank_start = args.dp_rank_start dp_address = args.dp_address dp_rpc_port = args.dp_rpc_port vllm_start_port = args.vllm_start_port def run_command(visible_devices, dp_rank, vllm_engine_port): command = [ "bash", "./run_dp_template.sh", visible_devices, str(vllm_engine_port), str(dp_size), str(dp_rank), dp_address, dp_rpc_port, str(tp_size), ] subprocess.run(command, check=True) if __name__ == "__main__": template_path = "./run_dp_template.sh" if not os.path.exists(template_path): print(f"Template file {template_path} does not exist.") sys.exit(1) processes = [] num_cards = dp_size_local * tp_size for i in range(dp_size_local): dp_rank = dp_rank_start + i vllm_engine_port = vllm_start_port + i visible_devices = ",".join(str(x) for x in range(i * tp_size, (i + 1) * tp_size)) process = multiprocessing.Process(target=run_command, args=(visible_devices, dp_rank, vllm_engine_port)) processes.append(process) process.start() for process in processes: process.join()
参数说明:
|参数|类型|必填|默认值|描述|
|---------|----|--------|-------|-----------|
|`--dp-size`|int|是|-|数据并行大小(所有节点上 DP rank 的总数)。|
|`--tp-size`|int|否|1|每个 DP rank 内的张量并行大小。|
|`--dp-size-local`|int|否|(与 `--dp-size` 相同)|当前节点上的 DP rank 数量。如果未设置,默认为 `--dp-size`。|
|`--dp-rank-start`|int|否|0|本节点上数据并行 rank 的起始 rank 偏移量。|
|`--dp-address`|str|是|-|数据并行主节点的 IP 地址。|
|`--dp-rpc-port`|str|否|12345|数据并行主节点通信的 RPC 端口。|
|`--vllm-start-port`|int|否|9000|本节点上每个 vLLM 引擎实例的起始端口。|
- 选择以下配置之一,并在每个节点上准备
run_dp_template.sh。
两个 Prefill 节点使用相同的模板。将 local_ip 设置为当前节点的 IP 地址。
Prefill 节点
nic_name="xxxx" # change to your own nic name
local_ip=xx.xx.xx.x # use the IP of the current Prefill node
export HCCL_OP_EXPANSION_MODE="AIV"
export HCCL_IF_IP=$local_ip
export GLOO_SOCKET_IFNAME=$nic_name
export TP_SOCKET_IFNAME=$nic_name
export HCCL_SOCKET_IFNAME=$nic_name
export VLLM_RPC_TIMEOUT=3600000
export VLLM_EXECUTE_MODEL_TIMEOUT_SECONDS=30000
export HCCL_EXEC_TIMEOUT=204
export HCCL_CONNECT_TIMEOUT=120
export OMP_PROC_BIND=false
export OMP_NUM_THREADS=10
export PYTORCH_NPU_ALLOC_CONF=expandable_segments:True
export HCCL_BUFFSIZE=1024
export TASK_QUEUE_ENABLE=1
export LD_PRELOAD=/usr/lib/aarch64-linux-gnu/libjemalloc.so.2:$LD_PRELOAD
export ASCEND_RT_VISIBLE_DEVICES=$1
# Ensure the model path matches the directory recorded during download
vllm serve /root/.cache/modelscope/hub/models/vllm-ascend/DeepSeek-V4-Pro-w4a8-mtp \
--host 0.0.0.0 \
--port $2 \
--data-parallel-address $5 \
--data-parallel-rpc-port $6 \
--data-parallel-size $3 \
--data-parallel-rank $4 \
--tensor-parallel-size $7 \
--enable-expert-parallel \
--seed 1024 \
--served-model-name dsv4 \
--max-model-len 131072 \
--max-num-batched-tokens 4096 \
--max-num-seqs 16 \
--gpu-memory-utilization 0.92 \
--block-size 128 \
--no-disable-hybrid-kv-cache-manager \
--tokenizer-mode deepseek_v4 \
--tool-call-parser deepseek_v4 \
--enable-auto-tool-choice \
--reasoning-parser deepseek_v4 \
--attention_config.indexer_kv_dtype int8 \
--model-loader-extra-config='{"enable_multithread_load": true, "num_threads": 128}' \
--trust-remote-code \
--quantization ascend \
--enforce-eager \
--speculative-config '{"num_speculative_tokens": 1,"method": "mtp","enforce_eager": true}' \
--additional-config '{"enable_cpu_binding":true,"enable_dsa_cp":true,"enable_fused_mc2":1,"enable_flashcomm1":true}' \
--kv-transfer-config \
'{"kv_connector": "MooncakeHybridConnector",
"kv_role": "kv_producer",
"kv_port": "30200",
"engine_id": "1",
"kv_connector_extra_config": {
"prefill": {
"dp_size": 2,
"tp_size": 16
},
"decode": {
"dp_size": 16,
"tp_size": 2
}
}
}'
两个 Decode 节点也使用同一个共享模板。将 local_ip 设置为当前节点的 IP 地址。
Decode 节点
nic_name="xxxx" # change to your own nic name
local_ip=xx.xx.xx.x # use the IP of the current Decode node
export LD_PRELOAD=/usr/lib/aarch64-linux-gnu/libjemalloc.so.2:$LD_PRELOAD
export HCCL_OP_EXPANSION_MODE="AIV"
export TASK_QUEUE_ENABLE=1
export VLLM_RPC_TIMEOUT=3600000
export VLLM_EXECUTE_MODEL_TIMEOUT_SECONDS=30000
export HCCL_EXEC_TIMEOUT=2000
export HCCL_CONNECT_TIMEOUT=1200
export HCCL_IF_IP=$local_ip
export GLOO_SOCKET_IFNAME=$nic_name
export TP_SOCKET_IFNAME=$nic_name
export HCCL_SOCKET_IFNAME=$nic_name
export OMP_PROC_BIND=false
export OMP_NUM_THREADS=10
export PYTORCH_NPU_ALLOC_CONF=expandable_segments:True
export HCCL_BUFFSIZE=1024
export ASCEND_RT_VISIBLE_DEVICES=$1
# Ensure the model path matches the directory recorded during download
vllm serve /root/.cache/modelscope/hub/models/vllm-ascend/DeepSeek-V4-Pro-w4a8-mtp \
--host 0.0.0.0 \
--port $2 \
--data-parallel-address $5 \
--data-parallel-rpc-port $6 \
--data-parallel-size $3 \
--data-parallel-rank $4 \
--tensor-parallel-size $7 \
--enable-expert-parallel \
--seed 1024 \
--served-model-name dsv4 \
--max-model-len 131072 \
--max-num-batched-tokens 120 \
--max-num-seqs 60 \
--gpu-memory-utilization 0.9 \
--block-size 128 \
--no-enable-prefix-caching \
--no-disable-hybrid-kv-cache-manager \
--tokenizer-mode deepseek_v4 \
--tool-call-parser deepseek_v4 \
--enable-auto-tool-choice \
--reasoning-parser deepseek_v4 \
--attention_config.indexer_kv_dtype int8 \
--model-loader-extra-config='{"enable_multithread_load": true, "num_threads": 128}' \
--trust-remote-code \
--quantization ascend \
--speculative-config '{"num_speculative_tokens": 1, "method":"mtp", "enforce_eager": true}' \
--compilation-config '{"cudagraph_mode": "FULL_DECODE_ONLY"}' \
--additional-config '{
"ascend_compilation_config":{
"enable_npugraph_ex":true,
"enable_static_kernel":false
},
"enable_cpu_binding":true,
"scheduler_config":{"recompute_scheduler_enable":true}
}' \
--kv-transfer-config \
'{"kv_connector": "MooncakeHybridConnector",
"kv_role": "kv_consumer",
"kv_port": "30800",
"engine_id": "8",
"kv_connector_extra_config": {
"prefill": {
"dp_size": 2,
"tp_size": 16
},
"decode": {
"dp_size": 16,
"tp_size": 2
}
}
}'
启动两个 Prefill 节点和两个 Decode 节点。两个 Prefill 命令都是必需的,因为每个节点拥有不同的 DP rank 范围。
```shell
# Prefill node 0
python launch_online_dp.py --dp-size 2 --tp-size 16 --dp-size-local 1 --dp-rank-start 0 --dp-address xx.xx.xx.1 --dp-rpc-port 12321 --vllm-start-port 7100
# Prefill node 1
python launch_online_dp.py --dp-size 2 --tp-size 16 --dp-size-local 1 --dp-rank-start 1 --dp-address xx.xx.xx.1 --dp-rpc-port 12321 --vllm-start-port 7100
# Decode node 0
python launch_online_dp.py --dp-size 16 --tp-size 2 --dp-size-local 8 --dp-rank-start 0 --dp-address xx.xx.xx.3 --dp-rpc-port 12321 --vllm-start-port 7100
# Decode node 1
python launch_online_dp.py --dp-size 16 --tp-size 2 --dp-size-local 8 --dp-rank-start 8 --dp-address xx.xx.xx.3 --dp-rpc-port 12321 --vllm-start-port 7100
```
=== "A3 系列与 DSpark" {#a3-series-with-dspark}
官方 `DeepSeek-V4-Pro-0813-w4a8` 检查点支持 DSpark,无需单独的 draft-model 路径。复用上面的 `launch_online_dp.py`,并使用以下按角色区分的模板。
**Prefill 节点**
```shell
nic_name="xxxx" # change to the NIC that owns local_ip
local_ip="xx.xx.xx.x" # use the IP of the current Prefill node
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib/
export LD_PRELOAD=/usr/lib/aarch64-linux-gnu/libjemalloc.so.2:$LD_PRELOAD
export HCCL_IF_IP=$local_ip
export GLOO_SOCKET_IFNAME=$nic_name
export TP_SOCKET_IFNAME=$nic_name
export HCCL_SOCKET_IFNAME=$nic_name
export VLLM_RPC_TIMEOUT=3600000
export VLLM_EXECUTE_MODEL_TIMEOUT_SECONDS=30000
export HCCL_TRANSFER_TIMEOUT=600
export HCCL_EXEC_TIMEOUT=204
export HCCL_CONNECT_TIMEOUT=6000
export OMP_PROC_BIND=false
export OMP_NUM_THREADS=10
export PYTORCH_NPU_ALLOC_CONF=expandable_segments:True
export HCCL_BUFFSIZE=1024
export TASK_QUEUE_ENABLE=1
export HCCL_OP_EXPANSION_MODE="AIV"
export VLLM_PREFIX_CACHE_RETENTION_INTERVAL=4096
export ASCEND_RT_VISIBLE_DEVICES=$1
# Ensure the model path matches the directory recorded during download
vllm serve /path/to/DeepSeek-V4-Pro-0813-w4a8 \
--host 0.0.0.0 \
--port $2 \
--data-parallel-address $5 \
--data-parallel-rpc-port $6 \
--data-parallel-size $3 \
--data-parallel-rank $4 \
--tensor-parallel-size $7 \
--enable-expert-parallel \
--seed 1024 \
--served-model-name dsv4 \
--max-model-len 150000 \
--max-num-batched-tokens 4096 \
--max-num-seqs 16 \
--gpu-memory-utilization 0.94 \
--block-size 32 \
--no-disable-hybrid-kv-cache-manager \
--tokenizer-mode deepseek_v4 \
--tool-call-parser deepseek_v4 \
--enable-auto-tool-choice \
--reasoning-parser deepseek_v4 \
--attention_config.indexer_kv_dtype int8 \
--model-loader-extra-config='{"enable_multithread_load": true, "num_threads": 128}' \
--trust-remote-code \
--quantization ascend \
--enforce-eager \
--speculative-config '{"num_speculative_tokens":5,"method":"dspark","enforce_eager":true}' \
--additional-config '{"enable_cpu_binding":true,"enable_dsa_cp":true,"enable_fused_mc2":1,"enable_flashcomm1":true}' \
--kv-transfer-config \
'{"kv_connector":"MooncakeHybridConnector",
"kv_role":"kv_producer",
"kv_port":"30100",
"engine_id":"1",
"kv_connector_extra_config":{
"prefill":{"dp_size":4,"tp_size":8},
"decode":{"dp_size":16,"tp_size":2}
}
}'
```
**Decode 节点**
```shell
nic_name="xxxx" # change to the NIC that owns local_ip
local_ip="xx.xx.xx.x" # use the IP of the current Decode node
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib/
export LD_PRELOAD=/usr/lib/aarch64-linux-gnu/libjemalloc.so.2:$LD_PRELOAD
export HCCL_IF_IP=$local_ip
export GLOO_SOCKET_IFNAME=$nic_name
export TP_SOCKET_IFNAME=$nic_name
export HCCL_SOCKET_IFNAME=$nic_name
export VLLM_RPC_TIMEOUT=3600000
export VLLM_EXECUTE_MODEL_TIMEOUT_SECONDS=30000
export HCCL_TRANSFER_TIMEOUT=600
export HCCL_EXEC_TIMEOUT=204
export HCCL_CONNECT_TIMEOUT=1200
export OMP_PROC_BIND=false
export OMP_NUM_THREADS=10
export PYTORCH_NPU_ALLOC_CONF=expandable_segments:True
export HCCL_BUFFSIZE=1800
export TASK_QUEUE_ENABLE=1
export ASCEND_RT_VISIBLE_DEVICES=$1
# Ensure the model path matches the directory recorded during download
vllm serve /path/to/DeepSeek-V4-Pro-0813-w4a8 \
--host 0.0.0.0 \
--port $2 \
--data-parallel-address $5 \
--data-parallel-rpc-port $6 \
--data-parallel-size $3 \
--data-parallel-rank $4 \
--tensor-parallel-size $7 \
--enable-expert-parallel \
--seed 1024 \
--served-model-name dsv4 \
--max-model-len 150000 \
--max-num-batched-tokens 96 \
--max-num-seqs 8 \
--gpu-memory-utilization 0.95 \
--block-size 32 \
--no-enable-prefix-caching \
--no-disable-hybrid-kv-cache-manager \
--tokenizer-mode deepseek_v4 \
--tool-call-parser deepseek_v4 \
--enable-auto-tool-choice \
--reasoning-parser deepseek_v4 \
--attention_config.indexer_kv_dtype int8 \
--model-loader-extra-config='{"enable_multithread_load": true, "num_threads": 128}' \
--trust-remote-code \
--quantization ascend \
--speculative-config '{"num_speculative_tokens":5,"method":"dspark","enforce_eager":true}' \
--compilation-config '{"cudagraph_mode":"FULL_DECODE_ONLY"}' \
--additional-config '{
"ascend_compilation_config": {
"enable_npugraph_ex": true,
"enable_static_kernel": false
},
"enable_cpu_binding": true,
"multistream_overlap_shared_expert": true,
"scheduler_config": {"recompute_scheduler_enable": true}
}' \
--kv-transfer-config \
'{"kv_connector":"MooncakeHybridConnector",
"kv_role":"kv_consumer",
"kv_port":"30800",
"engine_id":"8",
"kv_connector_extra_config":{
"prefill":{"dp_size":4,"tp_size":8},
"decode":{"dp_size":16,"tp_size":2}
}
}'
```
启动两个 Prefill 节点和两个 Decode 节点。分别将地址替换为 Prefill 和 Decode 主节点 IP。
```shell
# Prefill node 0
python launch_online_dp.py --dp-size 4 --tp-size 8 --dp-size-local 2 --dp-rank-start 0 --dp-address xx.xx.xx.1 --dp-rpc-port 12321 --vllm-start-port 7100
# Prefill node 1
python launch_online_dp.py --dp-size 4 --tp-size 8 --dp-size-local 2 --dp-rank-start 2 --dp-address xx.xx.xx.1 --dp-rpc-port 12321 --vllm-start-port 7100
# Decode node 0
python launch_online_dp.py --dp-size 16 --tp-size 2 --dp-size-local 8 --dp-rank-start 0 --dp-address xx.xx.xx.3 --dp-rpc-port 12321 --vllm-start-port 7100
# Decode node 1
python launch_online_dp.py --dp-size 16 --tp-size 2 --dp-size-local 8 --dp-rank-start 8 --dp-address xx.xx.xx.3 --dp-rpc-port 12321 --vllm-start-port 7100
```
-
部署 P-D 分离代理。
参考 Prefill-Decode 分离(Deepseek) 部署 P-D 分离代理。
5.2.2 A2 系列 PD 分离部署¶
本节展示在 Atlas 800 A2(64GB × 8)上部署一个 Prefill 池和一个 Decode 池。示例中每个池使用四个物理节点。
开始前,请:
-
在每个节点上准备脚本
launch_online_dp.py。import argparse import multiprocessing import os import subprocess import sys def parse_args(): parser = argparse.ArgumentParser() parser.add_argument( "--dp-size", type=int, required=True, help="Data parallel size." ) parser.add_argument( "--tp-size", type=int, default=1, help="Tensor parallel size." ) parser.add_argument( "--dp-size-local", type=int, default=-1, help="Local data parallel size." ) parser.add_argument( "--dp-rank-start", type=int, default=0, help="Starting rank for data parallel." ) parser.add_argument( "--dp-address", type=str, required=True, help="IP address for data parallel master node." ) parser.add_argument( "--dp-rpc-port", type=str, default=12345, help="Port for data parallel master node." ) parser.add_argument( "--vllm-start-port", type=int, default=9000, help="Starting port for the engine." ) return parser.parse_args() args = parse_args() dp_size = args.dp_size tp_size = args.tp_size dp_size_local = args.dp_size_local if dp_size_local == -1: dp_size_local = dp_size dp_rank_start = args.dp_rank_start dp_address = args.dp_address dp_rpc_port = args.dp_rpc_port vllm_start_port = args.vllm_start_port def run_command(visible_devices, dp_rank, vllm_engine_port): command = [ "bash", "./run_dp_template.sh", visible_devices, str(vllm_engine_port), str(dp_size), str(dp_rank), dp_address, dp_rpc_port, str(tp_size), ] subprocess.run(command, check=True) if __name__ == "__main__": template_path = "./run_dp_template.sh" if not os.path.exists(template_path): print(f"Template file {template_path} does not exist.") sys.exit(1) processes = [] num_cards = dp_size_local * tp_size for i in range(dp_size_local): dp_rank = dp_rank_start + i vllm_engine_port = vllm_start_port + i visible_devices = ",".join(str(x) for x in range(i * tp_size, (i + 1) * tp_size)) process = multiprocessing.Process(target=run_command, args=(visible_devices, dp_rank, vllm_engine_port)) processes.append(process) process.start() for process in processes: process.join() -
在每个节点上准备脚本
run_dp_template.sh。-
Prefill 节点(4 个 P 节点共用同一脚本)
nic_name="xxxx" # change to your own nic name local_ip=xx.xx.xx.1/2/3/4 # change to your own ip export LD_PRELOAD=/usr/lib/aarch64-linux-gnu/libjemalloc.so.2:$LD_PRELOAD export HCCL_OP_EXPANSION_MODE="AIV" export TASK_QUEUE_ENABLE=1 export VLLM_RPC_TIMEOUT=3600000 export VLLM_EXECUTE_MODEL_TIMEOUT_SECONDS=30000 export HCCL_EXEC_TIMEOUT=204 export HCCL_CONNECT_TIMEOUT=1200 export HCCL_IF_IP=$local_ip export GLOO_SOCKET_IFNAME=$nic_name export TP_SOCKET_IFNAME=$nic_name export HCCL_SOCKET_IFNAME=$nic_name export OMP_PROC_BIND=false export OMP_NUM_THREADS=10 export PYTORCH_NPU_ALLOC_CONF=expandable_segments:True export HCCL_BUFFSIZE=1024 sysctl -w vm.swappiness=0 sysctl -w kernel.numa_balancing=0 sysctl kernel.sched_migration_cost_ns=50000 export ASCEND_RT_VISIBLE_DEVICES=$1 # Ensure the model path matches the directory recorded during download vllm serve /root/.cache/modelscope/hub/models/vllm-ascend/DeepSeek-V4-Pro-w4a8-mtp \ --host 0.0.0.0 \ --port $2 \ --data-parallel-address $5 \ --data-parallel-rpc-port $6 \ --data-parallel-size $3 \ --data-parallel-rank $4 \ --tensor-parallel-size $7 \ --enable-expert-parallel \ --seed 1024 \ --served-model-name dsv4 \ --max-model-len 131072 \ --max-num-batched-tokens 4096 \ --max-num-seqs 16 \ --gpu-memory-utilization 0.9 \ --no-enable-prefix-caching \ --no-disable-hybrid-kv-cache-manager \ --tokenizer-mode deepseek_v4 \ --tool-call-parser deepseek_v4 \ --enable-auto-tool-choice \ --reasoning-parser deepseek_v4 \ --attention_config.indexer_kv_dtype int8 \ --model-loader-extra-config='{"enable_multithread_load": true, "num_threads": 128}' \ --trust-remote-code \ --quantization ascend \ --enforce-eager \ --speculative-config '{"num_speculative_tokens": 1, "method":"mtp", "enforce_eager": true}' \ --additional-config '{"enable_cpu_binding":true,"enable_shared_expert_dp":true,"enable_dsa_cp":true,"enable_flashcomm1":true}' \ --kv-transfer-config \ '{"kv_connector": "MooncakeHybridConnector", "kv_role": "kv_producer", "kv_port": "30000", "engine_id": "0", "kv_connector_extra_config": { "prefill": { "dp_size": 4, "tp_size": 8 }, "decode": { "dp_size": 8, "tp_size": 4 } } }'
-
-
解码节点(4 个 D 节点共用同一脚本)
```shell nic_name="xxxx" # change to your own nic name local_ip=xx.xx.xx.5/6/7/8 # change to your own ip export LD_PRELOAD=/usr/lib/aarch64-linux-gnu/libjemalloc.so.2:$LD_PRELOAD export HCCL_OP_EXPANSION_MODE="AIV" export TASK_QUEUE_ENABLE=1 export VLLM_RPC_TIMEOUT=3600000 export VLLM_EXECUTE_MODEL_TIMEOUT_SECONDS=30000 export HCCL_EXEC_TIMEOUT=204 export HCCL_CONNECT_TIMEOUT=1200 export HCCL_IF_IP=$local_ip export GLOO_SOCKET_IFNAME=$nic_name export TP_SOCKET_IFNAME=$nic_name export HCCL_SOCKET_IFNAME=$nic_name export OMP_PROC_BIND=false export OMP_NUM_THREADS=10 export PYTORCH_NPU_ALLOC_CONF=expandable_segments:True export HCCL_BUFFSIZE=1024 sysctl -w vm.swappiness=0 sysctl -w kernel.numa_balancing=0 sysctl kernel.sched_migration_cost_ns=50000 export ASCEND_RT_VISIBLE_DEVICES=$1 # Ensure the model path matches the directory recorded during download vllm serve /root/.cache/modelscope/hub/models/vllm-ascend/DeepSeek-V4-Pro-w4a8-mtp \ --host 0.0.0.0 \ --port $2 \ --data-parallel-address $5 \ --data-parallel-rpc-port $6 \ --data-parallel-size $3 \ --data-parallel-rank $4 \ --tensor-parallel-size $7 \ --enable-expert-parallel \ --seed 1024 \ --served-model-name dsv4 \ --max-model-len 131072 \ --max-num-batched-tokens 120 \ --max-num-seqs 60 \ --gpu-memory-utilization 0.9 \ --block-size 128 \ --no-enable-prefix-caching \ --no-disable-hybrid-kv-cache-manager \ --tokenizer-mode deepseek_v4 \ --tool-call-parser deepseek_v4 \ --enable-auto-tool-choice \ --reasoning-parser deepseek_v4 \ --attention_config.indexer_kv_dtype int8 \ --model-loader-extra-config='{"enable_multithread_load": true, "num_threads": 128}' \ --trust-remote-code \ --quantization ascend \ --speculative-config '{"num_speculative_tokens": 1, "method":"mtp", "enforce_eager": true}' \ --compilation-config '{"cudagraph_mode": "FULL_DECODE_ONLY"}' \ --additional-config '{"ascend_compilation_config":{"enable_npugraph_ex":true,"enable_static_kernel":false}, "enable_cpu_binding":true, "scheduler_config":{"recompute_scheduler_enable":true}}' \ --kv-transfer-config \ '{"kv_connector": "MooncakeHybridConnector", "kv_role": "kv_consumer", "kv_port": "30100", "engine_id": "1", "kv_connector_extra_config": { "prefill": { "dp_size": 4, "tp_size": 8 }, "decode": { "dp_size": 8, "tp_size": 4 } } }' ``` -
在每台节点上使用以下命令启动服务器。
-
预填充节点 0
-
预填充节点 1
-
预填充节点 2
-
预填充节点 3
-
解码节点 0
-
解码节点 1
-
解码节点 2
-
解码节点 3
-
-
部署 P-D 分离代理。
请参考 预填充-解码分离(Deepseek) 部署 P-D 分离代理。
关键参数说明:
--no-disable-hybrid-kv-cache-manager保持混合 KV 缓存管理器处于启用状态。DeepSeek-V4 KV Pool 部署需要此标志;否则,服务在启动时可能出现 OOM。--enforce-eager在预填充节点上强制使用 eager 执行,而不是图编译。--trust-remote-code允许加载模型仓库中的自定义代码。仅使用受信任的模型仓库。enable_dsa_cp: true在预填充节点上启用 DSA 上下文并行。DSA-CP 依赖 FlashComm1,当 DSA-CP 开启时会自动启用,因此无需在同一个--additional-config对象中设置"enable_flashcomm1": true。--kv-transfer-config配置 PD 分离中预填充生产者与解码消费者之间的 KV 缓存传输。kv_connector_extra_config.prefill.dp_size/tp_size和decode.dp_size/tp_size必须与预填充侧和解码侧实际的全局 DP 和 TP 布局相匹配。additional_config.enable_fused_mc2=1:启用 Fused MC2 融合算子,以加速预填充节点(A3 系列)上的通信。scheduler_config.recompute_scheduler_enable: true:启用重计算调度器。当解码节点的 KV 缓存不足时,请求会被发送到预填充节点进行 KV 缓存重计算。仅在 PD 解码节点上启用;旧版顶层键已弃用。--speculative-config:预填充节点和解码节点必须使用相同的 DSparknum_speculative_tokens值。请使用 checkpoint 中声明的值;本示例使用5。嵌套的enforce_eager适用于 DSpark 草稿执行,而顶层--enforce-eager使预填充目标模型保持 eager 模式。MooncakeHybridConnector:用于 PD 分离的 KV 传输连接器,在预填充节点和解码节点之间传输 KV 缓存。
部署验证:
PD分离服务完全启动后,通过prefill主节点上的代理端口发送请求,以验证Prefill和Decode节点是否正常协作。代理验证方法请参考Prefill-Decode分离(Deepseek)。
常见问题提示:如果遇到PD分离部署问题,请参阅公共FAQ进行故障排除。
6 功能验证¶
服务器启动后,您可以使用输入提示词查询模型:
在
curl http://<node0_ip>:<port>/v1/chat/completions \
-H "Content-Type: application/json" \
-d '{
"model": "dsv4",
"messages": [
{
"role": "user",
"content": "Who are you?"
}
],
"max_tokens": 256,
"temperature": 0
}'
预期结果:
服务返回HTTP 200 OK,并返回包含choices字段的JSON响应。
7 精度评估¶
7.1 使用AISBench¶
-
详细信息请参考使用AISBench。
-
执行后即可获得结果。
| 数据集 | 版本 | 指标 | 模式 | vllm-api-general-chat | 备注 |
|---|---|---|---|---|---|
| GPQA | - | 准确率 | gen | 91.41 | 0831官方W4A8权重,启用DSpark |
| GPQA | - | 准确率 | gen | 89.90 | 2台Atlas 800 A3(128GB × 8) |
| GSM8K | - | 准确率 | gen | 96.21 | 2台Atlas 800 A3(128GB × 8) |
8 性能评估¶
8.1 使用AISBench¶
详细信息请参考使用AISBench进行性能评估。
8.2 使用vLLM Benchmark¶
更多详细信息请参考vllm benchmark。
9 性能调优¶
9.1 推荐配置¶
注意:以下配置基于第5章的部署示例和特定测试环境,仅供参考,并非全局最优。尤其是长上下文配置仅为约1M token输入的建议起点,必须根据实际输入/输出长度、前缀缓存命中率、并发、内存使用情况和部署环境进行验证。进一步调优请参考第9.2节。
表1:场景概览¶
*Total NPUs表示所有节点上使用的NPU总数。
| 场景 | 部署模式 | *Total NPUs | 权重版本 | 关键考量 |
|---|---|---|---|---|
| 短序列 / 高吞吐 | 1P1D部署(2个Prefill + 2个Decode A3节点) | 64(A3) | DeepSeek-V4-Pro-0813-w4a8 | 与长上下文配置相比,使用更高的DP和并发:Prefill使用DP4/TP8和16个序列,Decode使用DP16/TP2和8个序列。在Prefill上启用FUSED_MC2以优先保证吞吐量。 |
| 长上下文(约1M输入,参考) | 1P1D部署(2个Prefill + 2个Decode A3节点) | 64(A3) | DeepSeek-V4-Pro-0813-w4a8 | 与短序列配置相比,两侧均使用DP2/TP16,将两侧序列数减少到2,并将Decode批量token数减少到16,以便为1M token上下文预留内存。 |
表2:详细节点配置¶
| 场景 | 配置 | NPUs | TP | DP | 最大序列数 | 最大批量Token数 | 最大模型长度 | DSpark投机数 | FUSED_MC2 | EP开关 | FC / DSA-CP开关 |
|---|---|---|---|---|---|---|---|---|---|---|---|
| 短序列 / 高吞吐 | Prefill池 | 32(2个A3节点) | 8 | 4 | 16 | 4096 | 150000 | 5 | 开启 | 开启 | 开启 / 开启 |
| 短序列 / 高吞吐 | Decode池 | 32(2个A3节点) | 2 | 16 | 8 | 96 | 150000 | 5 | 关闭 | 开启 | 关闭 / 关闭 |
| 长上下文(约1M输入,参考) | Prefill池 | 32(2个A3节点) | 16 | 2 | 2 | 4096 | 1048576 | 5 | 关闭 | 开启 | 开启 / 关闭 |
| 长上下文(约1M输入,参考) | Decode池 | 32(2个A3节点) | 16 | 2 | 2 | 16 | 1048576 | 5 | 关闭 | 开启 | 关闭 / 关闭 |
FC / DSA-CP开关分别表示FlashComm1和DSA上下文并行开关。对于长上下文Prefill配置,请在--additional-config中设置"enable_flashcomm1": true和"enable_dsa_cp": false。在Decode上,将重计算配置为"scheduler_config": {"recompute_scheduler_enable": true}。保持角色特定的--kv-transfer-config作为最后一个参数,Prefill使用"kv_port": "30100"和"engine_id": "1",Decode使用"kv_port": "30800"和"engine_id": "8"。在长上下文参考配置中禁用了FUSED_MC2,因为它会增加内存使用。如果内存充足且优先追求更高性能,建议在Prefill上启用FUSED_MC2。
--max-model-len 1048576是提示和生成token的总上下文预算,并不是在1M token提示之上额外的输出预算。如果内存不足,请减少--max-num-seqs或实际输入长度。完整的启动命令和参数说明请参考第5章中的部署示例。
9.2 调优指南¶
9.2.1 通用调优参考¶
调优方法请参考公共性能调优文档。
详细功能说明请参考功能矩阵。
10 常见问题¶
针对常见的环境、安装和通用参数问题,请参考公共FAQ;本章仅涵盖模型相关问题。
本页原文 Markdown:在 AtomGit 查看·内容源自开源项目 vl/vllm-ascend