☰
openPangu-72B快速上手教程:5分钟在昇腾NPU上启动vllm-ascend 0.11.0离线推理
2026/10/17 10:22:23 网站建设 项目流程
  • 大模型
  • 模型推理服务
  • AI 应用

【免费下载链接】openPangu-R-72B-2512-vllm-ascend-0.11.0

Ascend-SACT开源项目提供openPangu-72B适配昇腾A2的vllm-ascend 0.11.0推理镜像,预装CANN 8.5.0.alpha002、torch-npu 2.7.1等套件,修改权重目录即可快速启动离线推理,支持后训练扩展,方便开发者在昇腾NPU上高效部署大模型应用。【此简介由AI生成】

项目地址:https://ai.gitcode.com/Ascend-SACT/openPangu-72B-vllm-ascend-0.11.0
点击查看免费下载

openPangu-72B 是专为昇腾 A2 NPU 打造的 vllm-ascend 0.11.0 离线推理镜像项目。它把 CANN 8.5.0.alpha002、torch-npu 2.7.1 等经过验证的套件全部预装进镜像,启动后修改权重目录即可离线推理,无需手动搭建复杂的依赖环境。

📦 这个镜像能帮你做什么?

对于要在昇腾 NPU 上部署大模型的同学来说,环境依赖一直是最大的坑。本镜像的核心价值就一句话:把版本对齐这件事做完了。

软件配置项版本信息说明
vllmv0.11.0核心推理引擎
vllm-ascendv0.11.0昇腾适配层
CANN8.5.0.alpha002昇腾 AI 计算架构
Python3.11.13运行环境
transformers4.57.1模型加载框架
torch_npu2.7.1.post2NPU 计算后端

镜像中还内置了可直接运行的vllm_test.py测试脚本,vllm / vllm-ascend 安装在/vllm-project/目录,需要时可以直接修改源码。

仓库提供多个镜像包,按需选择即可:

镜像包用途
vllm-ascend0110-openpangu72b.tar.gzopenPangu 72B 推理镜像(本教程主角)
pangu72b-verl-vllm-v0.0.1.tar.gz加装 verl 的后训练镜像
openpangu72b-mauserv012.tar.gz基于训练 Dockerfile 打包,体积小,带 ma-user

详细说明见 README.md。

第一步:加载并启动推理镜像

将镜像包上传到昇腾服务器后,先导入镜像,再启动容器。仓库自带的 docker_run.sh 已配好完整的 NPU 直通参数,可直接参考:

docker load -i vllm-ascend0110-openpangu72b.tar.gz export IMAGE=vllm-ascend0110-openpangu72b:0.0.5 export NAME=openPangu72b-vllm0110 docker run -it -d --net=host \ --name $NAME \ --shm-size=1g \ --privileged \ --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 /home/models:/home/models \ -it $IMAGE bash

💡关键参数解读:--device三个设备映射是 NPU 直通的核心;-v /home/models:/home/models用于挂载模型权重目录,保证宿主机和容器内路径一致。

进入容器:

docker exec -it openPangu72b-vllm0110 /bin/bash

第二步:修改权重目录,5分钟跑通离线推理

启动容器后,在当前目录就能看到vllm_test.py。只需两步即可开始离线推理:

  1. 检查脚本中的权重路径,指向你的模型目录;
  2. 用本仓库的 config.json 替换权重目录下的原始 config.json(该文件声明了PanguProMoEV2ForCausalLM架构及auto_map映射,是模型正确加载的关键)。

然后直接运行:

python vllm_test.py

看到模型正常输出的响应内容,即代表离线推理链路已经打通 🎉。

⚠️ 替换 config.json 这一步不要省略,否则模型架构无法被 transformers 正确识别。

进阶:一键启动 OpenAI 兼容推理服务

如果不止本地测试,还要对外提供 API 服务,可以直接参考 serve_niu.sh。它针对 4 卡 NPU 做了完整的张量并行 + 专家并行配置:

export VLLM_USE_V1=1 export ASCEND_RT_VISIBLE_DEVICES=0,1,2,3 export HCCL_OP_EXPANSION_MODE="AIV" export VLLM_ASCEND_ENABLE_TOPK_OPTIMIZE=1 export PYTORCH_NPU_ALLOC_CONF=expandable_segments:True vllm serve /home/models/hf_models/openPangu-R-72B-2512 \ --tensor-parallel-size 4 \ --enable-expert-parallel \ --gpu-memory-utilization 0.85 \ --max-model-len 131072 \ --block-size 128 \ --trust-remote-code \ --port 8088

几个值得注意的调优点:

参数取值作用
--tensor-parallel-size44 卡张量并行,分摊 72B 模型权重
--enable-expert-parallel—启用专家并行,适配 MoE 架构
--max-model-len131072支持 128K 长上下文
--compilation_configFULL_DECODE_ONLY仅对解码阶段做 CUDA Graph 加速
--reasoning-parser/--tool-call-parserpangu开启思考模式解析与工具调用解析

调用服务:一条 curl 发请求

服务启动后(默认端口 8088),即可用仓库提供的 curl.sh 发起 OpenAI 兼容请求:

curl http://0.0.0.0:8088/v1/chat/completions \ -H "Content-Type: application/json" \ -d '{ "model": "openpangu-72b-vllm-0113", "messages": [{"role": "user", "content": "介绍一下大海"}], "max_tokens": 2048, "chat_template_kwargs": {"think": true, "reasoning_effort": "low"} }'

think: true会开启推理思考模式,reasoning_effort可控制思考深度,方便快速验证模型能力。

🔍 常见问题速查

问题排查方向
NPU 设备识别不到检查--device三项映射与昇腾驱动是否安装,容器内执行npu-smi info验证
模型加载报架构错误确认已用仓库的 config.json 替换权重目录下的配置文件
显存(HBM)不足降低--gpu-memory-utilization或增大--tensor-parallel-size卡数
长文本截断检查--max-model-len是否覆盖实际输入长度

📁 仓库核心文件一览

  • 项目说明:README.md
  • 容器启动脚本:docker_run.sh
  • 在线服务启动脚本:serve_niu.sh
  • API 调用示例:curl.sh
  • 模型配置文件:config.json
  • MoE 模型实现源码:modeling_pangu_moe.py
  • 推理镜像包:vllm-ascend0110-openpangu72b.tar.gz

总结:openPangu-72B 镜像帮你省去了最耗时的环境对齐工作——加载镜像、挂载权重、替换 config.json、运行vllm_test.py,5 分钟即可在昇腾 NPU 上跑通 vllm-ascend 0.11.0 离线推理;需要对外服务时,再参考serve_niu.sh一键拉起 4 卡推理服务即可。

  • 大模型
  • 模型推理服务
  • AI 应用

【免费下载链接】openPangu-R-72B-2512-vllm-ascend-0.11.0

Ascend-SACT开源项目提供openPangu-72B适配昇腾A2的vllm-ascend 0.11.0推理镜像,预装CANN 8.5.0.alpha002、torch-npu 2.7.1等套件,修改权重目录即可快速启动离线推理,支持后训练扩展,方便开发者在昇腾NPU上高效部署大模型应用。【此简介由AI生成】

项目地址:https://ai.gitcode.com/Ascend-SACT/openPangu-72B-vllm-ascend-0.11.0
点击查看免费下载

相关推荐

上一篇:如何用深蓝词库转换工具打破输入法格式壁垒:30+格式互通完整指南
下一篇:微信聊天记录导出完整指南:离线归档与年度报告

创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

需要专业的网站建设服务?

联系我们获取免费的网站建设咨询和方案报价,让我们帮助您实现业务目标

立即咨询