SuperAPI AI 工具集

npu-docker-launcher

部署与运维

查看项目

更新于 10-15 17:19

技能介绍

用途:该技能用于在昇腾(Ascend)NPU服务器上为用户直接启动Docker容器,覆盖启动前的环境检查、镜像查询、需求收集、命令生成与确认、执行与状态回报。

主要能力:启动前必须检查三项环境——驱动器目录 /usr/local/Ascend/driver/ 是否存在,/dev/davinci* 与 npu-smi 是否可用,Docker是否运行;任一不满足则停止并提示。随后用 docker images 列出本地镜像并按任务推荐(pytorch/tensorflow、mindspore、推理类如 vllm/mindie、dev/devel 等)。收集容器名、特权模式、网络模式、NPU卡选择、挂载目录、工作目录、环境变量等配置,生成 docker run 命令让用户确认后再执行。

使用方式:默认命令结构为 docker run -it -d,含 --name、--privileged、--network host,始终挂载 -v /usr/local/Ascend/driver:/usr/local/Ascend/driver 与 -v /usr/local/sbin:/usr/local/sbin,末尾为 /bin/bash。默认特权模式与host网络;指定卡时改用 --device=/dev/davinci{i};推理服务可用 bridge 网络加端口映射。执行后以 docker ps 验证并给出进入、日志、停止、删除等操作说明。

适用场景:开发环境、单卡训练、多卡分布式训练、带端口映射的推理服务。

限制:需已安装昇腾驱动与Docker,资料未提供具体收费或价格信息;未确认的镜像名、环境变量取值等需用户提供,技能不代为编造。