Atlas 300I DUO / Atlas 200I Pro¶
本教程介绍在昇腾310P推理硬件上部署vLLM,包括 Atlas 300I DUO和Atlas 200I Pro加速模块。
Note
Atlas 300I DUO不支持triton或triton-ascend。
在Atlas 300I DUO上运行vLLM¶
安装说明¶
如果从源码安装,vllm和vllm-ascend可能会自动引入triton和triton-ascend依赖,这在Atlas 300I DUO上可能导致意外问题。请在Atlas 300I DUO上运行前卸载它们:
图模式说明¶
Warning
当前版本在Atlas 300I DUO设备上支持FULL_DECODE_ONLY图模式,但由于硬件事件ID资源限制,存在以下限制:
- 当启用多个张量并行(TP)等级时,可捕获的图数量受限,具体取决于模型深度。例如,Qwen3-32B可以捕获并重放2个图。
- 当TP=1时,没有此限制。
- 我们已联系相关专家寻求解决方案。基于软件的修复被认为是可行的,但完全支持需要额外时间。感谢您的理解。
部署¶
根据您的硬件形态选择启动命令。
Atlas 300I DUO¶
运行Docker容器:
# Use the vllm-ascend image
export IMAGE=quay.io/ascend/vllm-ascend:v0.22.1rc1-310p
docker run --rm \
--name vllm-ascend \
--shm-size=10g \
--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/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 /root/.cache:/root/.cache \
-p 8080:8080 \
-it $IMAGE bash
Atlas 200I Pro加速模块¶
在Atlas 200I Pro上启动容器时,请挂载npu-smi所需的额外驱动库和
配置文件。如果没有这些挂载,容器内的npu-smi命令
可能会失败。请根据您的容器操作系统使用以下命令。
Note
Atlas 200I Pro也使用昇腾310P。请根据您要使用的NPU ID调整
--device=/dev/davinci0。您可以通过
ll /dev/ | grep davinci查询可用设备。
export IMAGE=quay.io/ascend/vllm-ascend:v0.22.1rc1-310p
docker run --rm \
--privileged \
--name vllm-ascend \
--shm-size=10g \
--device=/dev/davinci0:/dev/davinci0 \
--device=/dev/davinci_manager \
--device=/dev/ascend_manager \
--device=/dev/user_config \
-v /etc/sys_version.conf:/etc/sys_version.conf \
-v /etc/ld.so.conf.d/mind_so.conf:/etc/ld.so.conf.d/mind_so.conf \
-v /etc/hdcBasic.cfg:/etc/hdcBasic.cfg \
-v /var/dmp_daemon:/var/dmp_daemon \
-v /usr/lib64/libmmpa.so:/usr/lib64/libmmpa.so \
-v /usr/lib64/libcrypto.so.1.1:/usr/lib64/libcrypto.so.1.1 \
-v /usr/local/sbin/npu-smi:/usr/local/sbin/npu-smi \
-v /usr/lib64/libstackcore.so:/usr/lib64/libstackcore.so \
-v /usr/lib/aarch64-linux-gnu/libyaml-0.so.2:/usr/lib64/libyaml-0.so.2 \
-v /etc/slog.conf:/etc/slog.conf \
-v /var/slogd:/var/slogd \
-v /usr/local/Ascend/driver/lib64:/usr/local/Ascend/driver/lib64 \
-v /usr/lib64/libtensorflow.so:/usr/lib64/libtensorflow.so \
-v /root/.cache:/root/.cache \
-p 8080:8080 \
-it $IMAGE bash
export IMAGE=quay.io/ascend/vllm-ascend:v0.22.1rc1-310p-openeuler
docker run --rm \
--privileged \
--name vllm-ascend \
--shm-size=10g \
--device=/dev/davinci0:/dev/davinci0 \
--device=/dev/davinci_manager \
--device=/dev/ascend_manager \
--device=/dev/user_config \
-v /etc/sys_version.conf:/etc/sys_version.conf \
-v /etc/ld.so.conf.d/mind_so.conf:/etc/ld.so.conf.d/mind_so.conf \
-v /etc/hdcBasic.cfg:/etc/hdcBasic.cfg \
-v /var/dmp_daemon:/var/dmp_daemon \
-v /usr/lib64/libsemanage.so.2:/usr/lib64/libsemanage.so.2 \
-v /usr/lib64/libmmpa.so:/usr/lib64/libmmpa.so \
-v /usr/lib64/libcrypto.so.1.1:/usr/lib64/libcrypto.so.1.1 \
-v /usr/lib64/libyaml-0.so.2.0.9:/usr/lib64/libyaml-0.so.2 \
-v /usr/local/sbin/npu-smi:/usr/local/sbin/npu-smi \
-v /usr/lib64/libstackcore.so:/usr/lib64/libstackcore.so \
-v /etc/slog.conf:/etc/slog.conf \
-v /var/slogd:/var/slogd \
-v /usr/local/Ascend/driver/lib64:/usr/local/Ascend/driver/lib64 \
-v /usr/lib64/libtensorflow.so:/usr/lib64/libtensorflow.so \
-v /root/.cache:/root/.cache \
-p 8080:8080 \
-it $IMAGE bash
宿主机页缓存清理(推荐)¶
Tip
在Atlas 200I Pro上,如果宿主机内存紧张,或您之前运行过推理任务,可在**宿主机**上(需要root权限)于启动vllm serve之前考虑执行以下命令,以释放可回收的内核页缓存。这可能有助于释放宿主机内存,并降低模型加载时的OOM风险。该命令不会删除用户文件。
是否执行此清理取决于您的环境。请检查可用宿主机内存(例如使用free -h),并根据实际情况自行判断。如果内存已充足,可跳过此步骤。
设置环境变量:
NPU上的在线推理¶
Warning
对于Atlas 300I DUO(310P),不要依赖max-model-len自动检测
(即不要省略--max-model-len参数),因为这可能导致OOM。
原因,基于当前的310P注意力路径:
AscendAttentionMetadataBuilder310将model_config.max_model_len传递给AttentionMaskBuilder310。AttentionMaskBuilder310构建一个形状为[max_model_len, max_model_len]的float16完整因果掩码,然后将其转换为FRACTAL_NZ。- 在310P
attention_v1预填充/分块预填充路径 (_npu_flash_attention/_npu_paged_attention_splitfuse)中, 此显式掩码张量被直接消费,目前 没有压缩掩码路径。
如果自动检测解析出较大的上下文长度,掩码分配
(O(max_model_len^2))可能会超出NPU内存并触发OOM。
始终设置一个明确且保守的值,例如--max-model-len 16384。
运行以下命令,在NPU上为Qwen3 Dense系列启动vLLM服务器。
准备模型权重¶
使用来自Eco-Tech官方ModelScope仓库的W8A8SC量化权重。
| 模型 | ModelScope 链接 |
|---|---|
| Qwen3-8B-W8A8SC-310 | Eco-Tech/Qwen3-8B-w8a8sc-310-vllm |
| Qwen3-14B-W8A8SC-310 | Eco-Tech/Qwen3-14B-w8a8sc-310-vllm |
| Qwen3-32B-W8A8SC-310 | Eco-Tech/Qwen3-32B-w8a8sc-310-vllm |
vllm serve Eco-Tech/Qwen3-8B-w8a8sc-310-vllm/TP1/Qwen3-8B-w8a8sc-310-vllm-tp1 \
--host 127.0.0.1 \
--port 8080 \
--tensor-parallel-size 1 \
--gpu_memory_utilization 0.90 \
--max_num_seqs 32 \
--served_model_name qwen \
--dtype float16 \
--additional-config '{"ascend_compilation_config": {"fuse_norm_quant": false}}' \
--compilation-config '{"cudagraph_mode": "FULL_DECODE_ONLY", "cudagraph_capture_sizes": [1,2,4,8,16,32]}' \
--quantization ascend \
--max_model_len 16384 \
--no-enable-prefix-caching \
--load_format sharded_state
vllm serve Eco-Tech/Qwen3-14B-w8a8sc-310-vllm/TP1/Qwen3-14B-w8a8sc-310-vllm-tp1 \
--host 127.0.0.1 \
--port 8080 \
--tensor-parallel-size 1 \
--gpu_memory_utilization 0.90 \
--max_num_seqs 16 \
--served_model_name qwen \
--dtype float16 \
--additional-config '{"ascend_compilation_config": {"fuse_norm_quant": false}}' \
--compilation-config '{"cudagraph_mode": "FULL_DECODE_ONLY", "cudagraph_capture_sizes": [1,2,4,8,16]}' \
--quantization ascend \
--max_model_len 16384 \
--no-enable-prefix-caching \
--load_format sharded_state
export ASCEND_RT_VISIBLE_DEVICES=0,1,2,3
vllm serve Eco-Tech/Qwen3-32B-w8a8sc-310-vllm/TP4/Qwen3-32B-w8a8sc-310-vllm-tp4 \
--host 127.0.0.1 \
--port 8080 \
--tensor-parallel-size 4 \
--gpu_memory_utilization 0.90 \
--max_num_seqs 32 \
--served_model_name qwen \
--dtype float16 \
--additional-config '{"ascend_compilation_config": {"fuse_norm_quant": false}}' \
--compilation-config '{"cudagraph_mode": "FULL_DECODE_ONLY", "cudagraph_capture_sizes": [16,32]}' \
--quantization ascend \
--max_model_len 20480 \
--no-enable-prefix-caching \
--load_format sharded_state
服务器启动后,您可以使用输入提示查询模型:
curl http://localhost:8080/v1/completions \
-H "Content-Type: application/json" \
-d '{
"prompt": "The future of AI is",
"max_completion_tokens": 64,
"temperature": 0.0
}'
如果脚本运行成功,您可以看到生成的结果。
离线推理¶
运行以下脚本example.py,在NPU上执行离线推理。
import gc
import torch
from vllm import LLM, SamplingParams
from vllm.distributed.parallel_state import (
destroy_distributed_environment,
destroy_model_parallel,
)
def clean_up():
destroy_model_parallel()
destroy_distributed_environment()
gc.collect()
torch.npu.empty_cache()
prompts = [
"Hello, my name is",
"The future of AI is",
]
sampling_params = SamplingParams(
max_completion_tokens=100,
temperature=0.0,
)
llm = LLM(
model="Eco-Tech/Qwen3-8B-w8a8sc-310-vllm/TP1/Qwen3-8B-w8a8sc-310-vllm-tp1",
tensor_parallel_size=1,
max_model_len=16384,
dtype="float16",
quantization="ascend",
load_format="sharded_state",
additional_config={
"ascend_compilation_config": {
"fuse_norm_quant": False,
}
},
compilation_config={
"cudagraph_mode": "FULL_DECODE_ONLY",
"cudagraph_capture_sizes": [1, 2, 4, 8, 16, 32],
},
enable_prefix_caching=False,
)
outputs = llm.generate(prompts, sampling_params)
for output in outputs:
prompt = output.prompt
generated_text = output.outputs[0].text
print(f"Prompt: {prompt!r}, Generated text: {generated_text!r}")
del llm
clean_up()
import gc
import torch
from vllm import LLM, SamplingParams
from vllm.distributed.parallel_state import (
destroy_distributed_environment,
destroy_model_parallel,
)
def clean_up():
destroy_model_parallel()
destroy_distributed_environment()
gc.collect()
torch.npu.empty_cache()
prompts = [
"Hello, my name is",
"The future of AI is",
]
sampling_params = SamplingParams(
max_completion_tokens=100,
temperature=0.0,
)
llm = LLM(
model="Eco-Tech/Qwen3-14B-w8a8sc-310-vllm/TP1/Qwen3-14B-w8a8sc-310-vllm-tp1",
tensor_parallel_size=1,
max_model_len=16384,
dtype="float16",
quantization="ascend",
load_format="sharded_state",
additional_config={
"ascend_compilation_config": {
"fuse_norm_quant": False,
}
},
compilation_config={
"cudagraph_mode": "FULL_DECODE_ONLY",
"cudagraph_capture_sizes": [1, 2, 4, 8, 16],
},
enable_prefix_caching=False,
)
outputs = llm.generate(prompts, sampling_params)
for output in outputs:
prompt = output.prompt
generated_text = output.outputs[0].text
print(f"Prompt: {prompt!r}, Generated text: {generated_text!r}")
del llm
clean_up()
import gc
import os
import torch
from vllm import LLM, SamplingParams
from vllm.distributed.parallel_state import (
destroy_distributed_environment,
destroy_model_parallel,
)
def clean_up():
destroy_model_parallel()
destroy_distributed_environment()
gc.collect()
torch.npu.empty_cache()
os.environ["ASCEND_RT_VISIBLE_DEVICES"] = "0,1,2,3"
prompts = [
"Hello, my name is",
"The future of AI is",
]
sampling_params = SamplingParams(
max_completion_tokens=100,
temperature=0.0,
)
llm = LLM(
model="Eco-Tech/Qwen3-32B-w8a8sc-310-vllm/TP4/Qwen3-32B-w8a8sc-310-vllm-tp4",
tensor_parallel_size=4,
max_model_len=20480,
dtype="float16",
quantization="ascend",
load_format="sharded_state",
additional_config={
"ascend_compilation_config": {
"fuse_norm_quant": False,
}
},
compilation_config={
"cudagraph_mode": "FULL_DECODE_ONLY",
"cudagraph_capture_sizes": [16, 32],
},
enable_prefix_caching=False,
)
outputs = llm.generate(prompts, sampling_params)
for output in outputs:
prompt = output.prompt
generated_text = output.outputs[0].text
print(f"Prompt: {prompt!r}, Generated text: {generated_text!r}")
del llm
clean_up()
运行脚本:
如果脚本运行成功,您可以看到生成的结果。
结束语¶
如需抢先体验Qwen3-MoE、Qwen3-VL,以及Qwen3.5和Qwen3.6的性能加速预览支持,请关注#7394以获取更新的部署指南。