网络
虚拟私有云 VPC
弹性公网IP EIP
弹性负载均衡 ELB
NAT网关 NAT
云专线 DC
虚拟专用网络 VPN
云连接 CC
VPC终端节点 VPCEP
企业路由器 ER
企业交换机 ESW
全球加速 GA
企业连接 EC
云原生应用网络 ANC
安全与合规
安全技术与应用
Web应用防火墙 WAF
企业主机安全 HSS
云防火墙 CFW
安全云脑 SecMaster
DDoS防护 AAD
数据加密服务 DEW
数据库安全服务 DBSS
云堡垒机 CBH
数据安全中心 DSC
云证书管理服务 CCM
威胁检测服务 MTD
认证测试中心 CTC
边缘安全 EdgeSec
应用中间件
微服务引擎 CSE
分布式消息服务Kafka版
分布式消息服务RabbitMQ版
分布式消息服务RocketMQ版
API网关 APIG
分布式缓存服务 DCS
多活高可用服务 MAS
事件网格 EG
管理与监管
统一身份认证服务 IAM
消息通知服务 SMN
云监控服务 CES
应用运维管理 AOM
应用性能管理 APM
云日志服务 LTS
云审计服务 CTS
标签管理服务 TMS
配置审计 Config
应用身份管理服务 OneAccess
资源访问管理 RAM
组织 Organizations
资源编排服务 RFS
优化顾问 OA
IAM 身份中心
云运维中心 COC
资源治理中心 RGC
解决方案
高性能计算 HPC
SAP
混合云灾备
开天工业工作台 MIW
Haydn解决方案工厂
数字化诊断治理专家服务
云生态
云商店
合作伙伴中心
华为云开发者学堂
华为云慧通差旅
开发与运维
软件开发生产线 CodeArts
需求管理 CodeArts Req
流水线 CodeArts Pipeline
代码检查 CodeArts Check
编译构建 CodeArts Build
部署 CodeArts Deploy
测试计划 CodeArts TestPlan
制品仓库 CodeArts Artifact
移动应用测试 MobileAPPTest
CodeArts IDE Online
开源镜像站 Mirrors
性能测试 CodeArts PerfTest
应用管理与运维平台 ServiceStage
云应用引擎 CAE
开源治理服务 CodeArts Governance
华为云Astro轻应用
CodeArts IDE
Astro工作流 AstroFlow
代码托管 CodeArts Repo
漏洞管理服务 CodeArts Inspector
联接 CodeArtsLink
软件建模 CodeArts Modeling
Astro企业应用 AstroPro
CodeArts 盘古助手
华为云Astro大屏应用
计算
弹性云服务器 ECS
Flexus云服务
裸金属服务器 BMS
云手机服务器 CPH
专属主机 DeH
弹性伸缩 AS
镜像服务 IMS
函数工作流 FunctionGraph
云耀云服务器(旧版)
VR云渲游平台 CVR
Huawei Cloud EulerOS
云化数据中心 CloudDC
网络
虚拟私有云 VPC
弹性公网IP EIP
弹性负载均衡 ELB
NAT网关 NAT
云专线 DC
虚拟专用网络 VPN
云连接 CC
VPC终端节点 VPCEP
企业路由器 ER
企业交换机 ESW
全球加速 GA
企业连接 EC
云原生应用网络 ANC
CDN与智能边缘
内容分发网络 CDN
智能边缘云 IEC
智能边缘平台 IEF
CloudPond云服务
安全与合规
安全技术与应用
Web应用防火墙 WAF
企业主机安全 HSS
云防火墙 CFW
安全云脑 SecMaster
DDoS防护 AAD
数据加密服务 DEW
数据库安全服务 DBSS
云堡垒机 CBH
数据安全中心 DSC
云证书管理服务 CCM
威胁检测服务 MTD
认证测试中心 CTC
边缘安全 EdgeSec
大数据
MapReduce服务 MRS
数据湖探索 DLI
表格存储服务 CloudTable
可信智能计算服务 TICS
推荐系统 RES
云搜索服务 CSS
数据可视化 DLV
数据接入服务 DIS
数据仓库服务 GaussDB(DWS)
数据治理中心 DataArts Studio
湖仓构建 LakeFormation
智能数据洞察 DataArts Insight
应用中间件
微服务引擎 CSE
分布式消息服务Kafka版
分布式消息服务RabbitMQ版
分布式消息服务RocketMQ版
API网关 APIG
分布式缓存服务 DCS
多活高可用服务 MAS
事件网格 EG
开天aPaaS
应用平台 AppStage
开天企业工作台 MSSE
开天集成工作台 MSSI
API中心 API Hub
云消息服务 KooMessage
交换数据空间 EDS
云地图服务 KooMap
云手机服务 KooPhone
组织成员账号 OrgID
云空间服务 KooDrive
管理与监管
统一身份认证服务 IAM
消息通知服务 SMN
云监控服务 CES
应用运维管理 AOM
应用性能管理 APM
云日志服务 LTS
云审计服务 CTS
标签管理服务 TMS
配置审计 Config
应用身份管理服务 OneAccess
资源访问管理 RAM
组织 Organizations
资源编排服务 RFS
优化顾问 OA
IAM 身份中心
云运维中心 COC
资源治理中心 RGC
区块链
区块链服务 BCS
数字资产链 DAC
华为云区块链引擎服务 HBS
解决方案
高性能计算 HPC
SAP
混合云灾备
开天工业工作台 MIW
Haydn解决方案工厂
数字化诊断治理专家服务
价格
成本优化最佳实践
专属云商业逻辑
云生态
云商店
合作伙伴中心
华为云开发者学堂
华为云慧通差旅
其他
管理控制台
消息中心
产品价格详情
系统权限
客户关联华为云合作伙伴须知
公共问题
宽限期保留期
奖励推广计划
活动
云服务信任体系能力说明
开发与运维
软件开发生产线 CodeArts
需求管理 CodeArts Req
流水线 CodeArts Pipeline
代码检查 CodeArts Check
编译构建 CodeArts Build
部署 CodeArts Deploy
测试计划 CodeArts TestPlan
制品仓库 CodeArts Artifact
移动应用测试 MobileAPPTest
CodeArts IDE Online
开源镜像站 Mirrors
性能测试 CodeArts PerfTest
应用管理与运维平台 ServiceStage
云应用引擎 CAE
开源治理服务 CodeArts Governance
华为云Astro轻应用
CodeArts IDE
Astro工作流 AstroFlow
代码托管 CodeArts Repo
漏洞管理服务 CodeArts Inspector
联接 CodeArtsLink
软件建模 CodeArts Modeling
Astro企业应用 AstroPro
CodeArts 盘古助手
华为云Astro大屏应用
存储
对象存储服务 OBS
云硬盘 EVS
云备份 CBR
高性能弹性文件服务 SFS Turbo
弹性文件服务 SFS
存储容灾服务 SDRS
云硬盘备份 VBS
云服务器备份 CSBS
数据快递服务 DES
云存储网关 CSG
专属分布式存储服务 DSS
数据工坊 DWR
地图数据 MapDS
键值存储服务 KVS
容器
云容器引擎 CCE
云容器实例 CCI
容器镜像服务 SWR
云原生服务中心 OSC
应用服务网格 ASM
华为云UCS
数据库
云数据库 RDS
数据复制服务 DRS
文档数据库服务 DDS
分布式数据库中间件 DDM
云数据库 GaussDB
云数据库 GeminiDB
数据管理服务 DAS
数据库和应用迁移 UGO
云数据库 TaurusDB
人工智能
AI开发平台ModelArts
华为HiLens
图引擎服务 GES
图像识别 Image
文字识别 OCR
自然语言处理 NLP
内容审核 Moderation
图像搜索 ImageSearch
医疗智能体 EIHealth
企业级AI应用开发专业套件 ModelArts Pro
人脸识别服务 FRS
对话机器人服务 CBS
语音交互服务 SIS
人证核身服务 IVS
视频智能分析服务 VIAS
城市智能体
自动驾驶云服务 Octopus
盘古大模型 PanguLargeModels
IoT物联网
设备接入 IoTDA
全球SIM联接 GSL
IoT数据分析 IoTA
路网数字化服务 DRIS
IoT边缘 IoTEdge
设备发放 IoTDP
企业应用
域名注册服务 Domains
云解析服务 DNS
企业门户 EWP
ICP备案
商标注册
华为云WeLink
华为云会议 Meeting
隐私保护通话 PrivateNumber
语音通话 VoiceCall
消息&短信 MSGSMS
云管理网络
SD-WAN 云服务
边缘数据中心管理 EDCM
云桌面 Workspace
应用与数据集成平台 ROMA Connect
ROMA资产中心 ROMA Exchange
API全生命周期管理 ROMA API
政企自服务管理 ESM
视频
实时音视频 SparkRTC
视频直播 Live
视频点播 VOD
媒体处理 MPC
视频接入服务 VIS
数字内容生产线 MetaStudio
迁移
主机迁移服务 SMS
对象存储迁移服务 OMS
云数据迁移 CDM
迁移中心 MGC
专属云
专属计算集群 DCC
开发者工具
SDK开发指南
API签名指南
DevStar
华为云命令行工具服务 KooCLI
Huawei Cloud Toolkit
CodeArts API
云化转型
云架构中心
云采用框架
用户服务
账号中心
费用中心
成本中心
资源中心
企业管理
工单管理
客户运营能力
国际站常见问题
支持计划
专业服务
合作伙伴支持计划
我的凭证
华为云公共事业服务云平台
工业软件
工业数字模型驱动引擎
硬件开发工具链平台云服务
工业数据转换引擎云服务

自动化脚本快速部署推理服务(推荐)

更新时间:2025-02-25 GMT+08:00
分享

场景描述

本方案提供了一键式安装脚本start.sh,用于快速部署推理服务。脚本中实现了以下步骤的自动化操作:

  1. 环境检查
  2. 拉取镜像
  3. 根据实际值更新rank_table_file.json
  4. 启动容器
  5. 进入容器启动服务

前提条件

  1. 已经完成资源购买
  2. 请确保多机之间能够正常通信。
  3. 请确保每台机器都能访问到西南-贵阳一SWR镜像仓库。

约束限制

脚本中的镜像是在西南-贵阳一区域,建议在西南-贵阳一区域上部署推理服务。

脚本默认只支持在标准的欧拉或HCE 2.0操作系统上执行。

步骤一:检查环境

  1. SSH登录机器后,检查NPU设备检查。如果驱动版本不是24.1.0,请先升级驱动和对应固件。
    npu-smi info -t board -i 1 | egrep -i "software|firmware"   #查看驱动和固件版本

    24.1.0版本驱动文件为Ascend-hdk-910b-npu-driver_24.1.0_linux-aarch64.run,对应固件文件为Ascend-hdk-910b-npu-firmware_7.5.0.3.220.run,请申请下载。

    安装固件命令如下,安装完后需要reboot重启机器。

    chmod 700 *.run
    ./Ascend-hdk-910b-npu-firmware_7.5.0.3.220.run --full
    reboot

    安装24.1.0驱动命令如下:

    ./Ascend-hdk-910b-npu-driver_24.1.0_linux-aarch64.run --full --install-for-all

    安装完成后再使用如下命令查看是否安装正确。

    npu-smi info -t board -i 1 | egrep -i "software|firmware"
  2. 检查docker是否安装。
    docker -v   #检查docker是否安装

    如尚未安装,运行以下命令安装docker。

    yum install -y docker-engine.aarch64 docker-engine-selinux.noarch docker-runc.aarch64
  3. 配置IP转发,用于容器内的网络访问。执行以下命令查看net.ipv4.ip_forward配置项的值,如果为1,可跳过此步骤。
    sysctl -p | grep net.ipv4.ip_forward
    如果net.ipv4.ip_forward配置项的值不为1,执行以下命令配置IP转发。
    sed -i 's/net\.ipv4\.ip_forward=0/net\.ipv4\.ip_forward=1/g' /etc/sysctl.conf
    sysctl -p | grep net.ipv4.ip_forward

步骤二:自动化部署

  1. 需要先安装yum和expect。
    sudo apt-get install yum
    yum install expect
  2. 把rank_table_file.json、start.sh和权重文件放在同一目录下。其中rank_table_file.json和start.sh只需要在主节点即可。rank_table_file.json仅作为模板使用,会在start.sh脚本里自动更新,文件内容详见附录:rank_table_file.json文件一键部署脚本start.sh。参考目录结构如下:
    ${mountPath}
    |---rank_table_file.json
    |---start.sh
    |---DeepSeekR1-w8a8  # 权重文件所在目录,每个节点都要有,且目录结构保持一致
  1. 执行部署脚本start.sh。
    sh start.sh --model_name ${modelName} --mount_path ${mountPath} --maxSeqLen ${maxSeqLen} --node_ips ${nodeIps} --passwords ${nodePwds}
    • modelName:模型权重所在文件夹名称。注意:文件夹名称需要包含R1或V3。例如:DeepSeekR1-w8a8。
    • mountPath:容器挂载的路径,且不能为/home,该路径下包含权重文件所在目录。即为权重文件所在目录的父目录。
    • maxSeqLen:输入长度+输出长度的最大值。推荐默认16384。
    • nodeIps:节点IP列表,使用“,“分隔。填2个节点IP地址。
    • nodePwds:各节点的root用户登录密码,使用“,“分隔,和上述节点要一一对应。

步骤三:调用

执行请求调用模型

curl -ik -H 'Content-Type: application/json' -d '{"messages":[{"role":"user","content":"请讲一个笑话"}],"model":"${model_name}","temperature":0.6,"max_tokens":1024}' -X POST http://${ip}:${port}/v1/chat/completions
  • model_name:为要调用的模型名称,即DeepSeek-V3或DeepSeek-R1
  • ip:为步骤二:自动化部署中nodeIps中第一个IP
  • port:为要访问的端口,默认1025

一键部署脚本start.sh

#!/bin/bash

# 定义默认值
# 参数变量
model_name=""
mount_path=""
maxSeqLen=
image_name="swr.cn-southwest-2.myhuaweicloud.com/ei_ascendcloud_devops/mindie:2.0.T3.1-800I-A2-py311-openeuler24.03-lts-0220"
CONTAINER_NAME="mindie"
# 根据变量修改的值
MODEL_NAME=""
model_path=""
mount_path="${mount_path%/}"
RANK_TABLE_FILE="$mount_path/rank_table_file.json"

port=22
# 默认maxSeqLen为16K时的参数配置,如果maxSeqLen修改为32K时,需要修改maxInputTokenLen、maxPrefillTokens和maxIterTimes的值,具体修改方法参见附录:config.json文件介绍。
maxPrefillBatchSize=100
maxInputTokenLen=8192
maxPrefillTokens=8192
maxIterTimes=8192
httpsEnabled=false
multiNodesInferEnabled=true
interCommTLSEnabled=false
interNodeTLSEnabled=false


# 定义期望的软件和固件版本
expected_software_version="24.1.0"
expected_firmware_version="7.5.0.3.220"


# 定义节点IP列表(第一个IP为主节点)
node_ips=("")
NODE_IPS=""
echo "node_ips:   ${node_ips[*]}"

# 定义密码列表(如果只有一个密码,则所有节点使用该密码)
# node_passwords=("xxx" "xxx")  # 替换为实际密码
node_passwords=("")


# 检查expect是否安装
if ! command -v expect &> /dev/null; then
    echo "错误: expect未安装,请先执行yum install expect"
    exit 1
fi

# 解析命令行参数
ARGS=$(getopt -o n:p:l:i:k: -l model_name:,mount_path:,maxSeqLen:,node_ips:,passwords: -- "$@")
if [ $? -ne 0 ]; then
    echo "参数解析失败"
    exit 1
fi

eval set -- "$ARGS"

while true; do
    case "$1" in
        -n|--model_name)
            model_name=$2  # 忽略大小写
            MODEL_NAME=$model_name
            shift 2
            ;;
        -p|--mount_path)
            mount_path="$2"
            # 去除 mount_path 最后的 '/'
            mount_path="${mount_path%/}"
            shift 2
            ;;
        -l|--maxSeqLen)
            maxSeqLen="$2"
            shift 2
            ;;
        -i|--node_ips)
            IFS=',' read -r -a node_ips <<< "$2"
            shift 2
            ;;
        -k|--passwords)
            IFS=',' read -r -a node_passwords <<< "$2"
            shift 2
            ;;
        --)
            shift
            break
            ;;
        *)
            echo "用法:$0 -n <model_name> -p <mount_path> -l <maxSeqLen> -i <node_ips> -k <passwords>"
            echo "  -n, --model-name: 模型名称,可选值为 r1 或 v3"
            echo "  -p, --mount-path: 挂载路径,必须提供,且不能为 /home"
            echo "  -l, --max-seq-len: 输入长度+输出长度的最大值,需要在8192上加2^的若干次方数"
            echo "  -i, --node-ips: 逗号分隔的节点IP地址"
            echo "  -k, --passwords: 密码字符串列表"
            exit 1
            ;;
    esac
done

# 检查 mount_path 是否为空或为 /home
if [ -z "$mount_path" ]; then
    echo "必须提供 mount_path 参数"
    exit 1
fi

if [ "$mount_path" = "/home" ]; then
    echo "mount_path 不能为 /home"
    exit 1
fi

if [ -z "$maxSeqLen" ]; then
    echo "必须提供 maxSeqLen 参数"
    exit 1
fi
model_path=$mount_path/$model_name
# 转换 model_name 为小写,以便忽略大小写
lower_model_name=$(echo "$model_name" | tr '[:upper:]' '[:lower:]')

# 检查 model_name 是否包含 r1 或 v3
if [[ "$lower_model_name" == *"r1"* ]]; then
    model_name="DeepSeek-R1"
    # model_path=$mount_path/deepseekR1-w8a8
    # echo "模型路径为:$model_path"

elif [[ "$lower_model_name" == *"v3"* ]]; then
    model_name="DeepSeek-V3"
    # model_path=$mount_path/deepseekV3-w8a8
    # echo "模型路径为:$model_path"
else
    echo "错误: model_name 参数中必须包含 r1 或 v3"
    exit 1
fi


RANK_TABLE_FILE="$mount_path/rank_table_file.json"
# 打印参数值
echo "maxSeqLen的值为:$maxSeqLen"
echo "模型名:$model_name"
echo "模型路径:$model_path"
echo "挂载路径:$mount_path"
echo "node_ips:   ${node_ips[*]}"
# 将数组元素以空格连接成字符串,再使用tr将空格替换为逗号
NODE_IPS=$(printf "%s " "${node_ips[@]}" | tr ' ' ',')
# 去除末尾多余的逗号
NODE_IPS=${NODE_IPS%,}
echo "NODE_IPS: ${NODE_IPS}"
echo "node_passwords:   ${node_passwords[*]}"
echo "rank_table_file文件路径:$RANK_TABLE_FILE"
# exit 0


echo "步骤1: 检查NPU的软件和固件信息..."
npu_smi_output=$(npu-smi info -t board -i 1 | egrep -i "software|firmware")

if [ $? -ne 0 ]; then
    echo "检查NPU软件和固件信息失败,退出脚本。"
    exit 1
fi


# 提取实际的软件和固件版本
actual_software_version=$(echo "$npu_smi_output" | grep -i "Software Version" | awk -F': ' '{print $2}' | tr -d ' ')
actual_firmware_version=$(echo "$npu_smi_output" | grep -i "Firmware Version" | awk -F': ' '{print $2}' | tr -d ' ')

# 比较软件版本
if echo -e "$actual_software_version\n$expected_software_version" | sort -V | head -n1 | grep -q "$actual_software_version"; then
    if [ "$actual_software_version" != "$expected_software_version" ]; then
        echo "软件版本不对,期望版本至少为 $expected_software_version,实际版本为 $actual_software_version"
        exit 1
    fi
fi

# 比较固件版本
if echo -e "$actual_firmware_version\n$expected_firmware_version" | sort -V | head -n1 | grep -q "$actual_firmware_version"; then
    if [ "$actual_firmware_version" != "$expected_firmware_version" ]; then
        echo "固件版本不对,期望版本至少为 $expected_firmware_version,实际版本为 $actual_firmware_version"
        exit 1
    fi
fi

echo "NPU软件和固件版本检查通过。"

# 步骤2: 执行一系列检查命令,如果有报错则退出
echo "步骤2: 执行检查命令..."
commands=(
    "for i in {0..7}; do hccn_tool -i \$i -lldp -g | grep Ifname; done"
    "for i in {0..7}; do hccn_tool -i \$i -link -g ; done"
    "for i in {0..7}; do hccn_tool -i \$i -net_health -g ; done"
    "for i in {0..7}; do hccn_tool -i \$i -netdetect -g ; done"
    "for i in {0..7}; do hccn_tool -i \$i -gateway -g ; done"
    "for i in {0..7}; do hccn_tool -i \$i -tls -g ; done | grep switch"
    "for i in {0..7}; do hccn_tool -i \$i -tls -s enable 0; done"
)

echo "拉取镜像"
docker pull $image_name

# 步骤4: 获取enp67s0f5网卡下的IP地址
echo "步骤4: 获取本机IP地址..."
ip_addr_now=$(hostname -I | awk '{print $1}')
if [ -z "$ip_addr_now" ]; then
    echo "获取当前物理机IP地址失败。"
    exit 1
fi
echo "获取当前物理机IP地址成功,IP地址为: $ip_addr_now"


# 初始化标记位
is_first_node=false
echo "主节点ip地址为:${node_ips[0]}"
# exit 0
# 判断 ip_addr_now 是否是第一个节点
if [ "$ip_addr_now" = "${node_ips[0]}" ]; then
    is_first_node=true
fi

if $is_first_node; then
    echo "${ip_addr_now}作为主节点"

    # 步骤3: 获取每张卡的IP地址并打印(修正版)
    echo "步骤3: 获取每张卡的IP地址..."

    # 获取主节点的显卡IP
    echo "步骤3: 获取主节点的显卡IP..."
    declare -a main_npu_ips
    for i in {0..7}; do
        ip_info=$(hccn_tool -i $i -ip -g 2>&1)
        if [ $? -ne 0 ]; then
            echo "获取卡号 $i 的IP地址失败,退出脚本。"
            exit 1
        fi

        ip_addr=$(echo "$ip_info" | awk -F':' '/ipaddr/ {print $2}' | tr -d ' ')
        if [ -z "$ip_addr" ]; then
            echo "卡号 $i 未找到有效IP地址,退出脚本。"
            exit 1
        fi
        main_npu_ips[$i]=$ip_addr
        echo "主节点NPU卡$i IP地址: ${main_npu_ips[$i]}"
    done
    echo "主节点IP地址获取完成。"

    # 备份原始文件
    cp "$RANK_TABLE_FILE" "${RANK_TABLE_FILE}.bak"

    # 设置环境变量传递显卡IP
    export ips_0="${main_npu_ips[@]}"

    # 更新主节点的server_id和container_ip (使用Python)
    python3 <<EOF
import json
import os

def update_json(filename, server_id, container_ip):
    device_ips = os.getenv("ips_0").split()
    with open(filename, 'r') as f:
        data = json.load(f)
    # 更新主节点的server_id和container_ip
    data['server_list'][0]['server_id'] = server_id
    data['server_list'][0]['container_ip'] = container_ip
    # 更新主节点的device_ip
    for i in range(8):
        if i < len(device_ips):
            data['server_list'][0]['device'][i]['device_ip'] = device_ips[i]
        else:
            data['server_list'][0]['device'][i]['device_ip'] = ""
    # 删除多余的设备条目(如果有的话)
    for server in data['server_list']:
        if len(server['device']) > 8:
            server['device'] = server['device'][:8]
    with open(filename, 'w') as f:
        json.dump(data, f, indent=4)

update_json("$RANK_TABLE_FILE", "${node_ips[0]}", "${node_ips[0]}")
EOF

    # 远程登录其他节点,获取显卡IP并更新rank_table_file.json
    for (( node_index=1; node_index<${#node_ips[@]}; node_index++ )); do
        node_ip=${node_ips[$node_index]}
        node_password=${node_passwords[$node_index]} 

        echo "步骤3: 获取节点 $node_ip 的显卡IP..."

        # 使用expect远程执行命令获取显卡IP
        declare -a remote_npu_ips
        for i in {0..7}; do
            ip_info=$(expect -c "
                set timeout 10
                spawn ssh -p $port -o StrictHostKeyChecking=no root@$node_ip \"hccn_tool -i $i -ip -g\"
                expect {
                    \"password:\" {
                        send \"${node_password}\r\"
                        exp_continue
                    }
                    eof {
                        catch wait result
                        exit [lindex \$result 3]
                    }
                }
            ")
            if [ $? -ne 0 ]; then
                echo "获取节点 $node_ip 卡号 $i 的IP地址失败,退出脚本。"
                exit 1
            fi

            ip_addr=$(echo "$ip_info" | awk -F':' '/ipaddr/ {print $2}' | tr -d ' ')
            if [ -z "$ip_addr" ]; then
                echo "节点 $node_ip 卡号 $i 未找到有效IP地址,退出脚本。"
                exit 1
            fi
            remote_npu_ips[$i]=$ip_addr
            echo "节点 $node_ip NPU卡$i IP地址: ${remote_npu_ips[$i]}"
        done
        echo "节点 $node_ip IP地址获取完成。"

        # 设置环境变量传递显卡IP
        export ips_1="${remote_npu_ips[@]}"

        # 更新当前节点的server_id和container_ip 
        python3 <<EOF
import json
import os

def update_server_info(filename, server_index, server_id, container_ip):
    device_ips = os.getenv("ips_1").split()
    with open(filename, 'r') as f:
        data = json.load(f)

    # 确保 server_list 有足够的条目
    while server_index >= len(data['server_list']):
        new_server = {
            "device": [
                {"device_id": str(i), "device_ip": "", "rank_id": str(len(data['server_list']) * 8 + i)} for i in range(8)
            ],
            "server_id": "",
            "container_ip": ""
        }
        data['server_list'].append(new_server)

    # 更新指定服务器的信息
    data['server_list'][server_index]['server_id'] = server_id
    data['server_list'][server_index]['container_ip'] = container_ip

    # 更新设备 IP 和 rank_id
    for i in range(min(len(device_ips), 8)):
        data['server_list'][server_index]['device'][i]['device_ip'] = device_ips[i]
        data['server_list'][server_index]['device'][i]['rank_id'] = str(server_index * 8 + i)

    # 确保设备条目数量正确
    data['server_list'][server_index]['device'] = data['server_list'][server_index]['device'][:8]

    # 更新 server_count
    data['server_count'] = str(len(data['server_list']))

    with open(filename, 'w') as f:
        json.dump(data, f, indent=4)

update_server_info("$RANK_TABLE_FILE", $node_index, "$node_ip", "$node_ip")
EOF
    done


    # 传输给子节点
    # 传输 rank_table_file.json 到其他节点
    for (( node_index=1; node_index<${#node_ips[@]}; node_index++ )); do
        node_ip=${node_ips[$node_index]}
        node_password=${node_passwords[$node_index]} 

        echo "传输 rank_table_file.json 到节点 $node_ip..."

        # 使用 expect 自动输入密码并执行 scp
        expect -c "
            set timeout 10
            spawn scp -P $port -ro StrictHostKeyChecking=no $mount_path/rank_table_file.json $mount_path/start.sh root@$node_ip:$mount_path
            expect {
                \"password:\" {
                    send \"$node_password\r\"
                    exp_continue
                }
                eof {
                    catch wait result
                    exit [lindex \$result 3]
                }
            }
        "

        if [ $? -eq 0 ]; then
            echo "文件传输到节点 $node_ip 成功。"
        else
            echo "文件传输到节点 $node_ip 失败。"
            exit 1
        fi
    done

    echo "所有文件传输完成。"

    # 连接到远程节点并执行命令
    for (( node_index=1; node_index<${#node_ips[@]}; node_index++ )); do
        node_ip=${node_ips[$node_index]}
        node_password=${node_passwords[$node_index]}  

        echo "连接到节点 $node_ip 并执行命令..."

        expect -c "
            set timeout 10
            spawn ssh -p $port -o StrictHostKeyChecking=no root@$node_ip \"bash $mount_path/start.sh -n $MODEL_NAME -p $mount_path -l $maxSeqLen -i $NODE_IPS > $mount_path/log_$node_ip.txt 2>&1\"
            expect {
                \"password:\" {
                    send \"$node_password\r\"
                    exp_continue
                }
                eof {
                    catch wait result
                    exit [lindex \$result 3]
                }
            }
        "

        if [ $? -eq 0 ]; then
            echo "命令在节点 $node_ip 执行成功。"
        else
            echo "命令在节点 $node_ip 执行失败。"
            exit 1
        fi
    done

    echo "远程执行脚本成功"
sleep 32
fi

# 步骤5: 启动容器
echo "步骤5: 启动容器..."

# 启动容器
docker run -itd --privileged \
    -v /etc/localtime:/etc/localtime \
    -v /usr/local/Ascend/driver:/usr/local/Ascend/driver \
    -v /usr/local/Ascend/firmware:/usr/local/Ascend/firmware \
    --device=/dev/davinci_manager \
    --device=/dev/devmm_svm \
    --device=/dev/hisi_hdc \
    -v /var/log/npu/:/usr/slog \
    -v /usr/local/sbin/npu-smi:/usr/local/sbin/npu-smi \
    -v /sys/fs/cgroup:/sys/fs/cgroup:ro \
    -v $mount_path:$mount_path \
    --net=host \
    --ipc=host \
    --shm-size 2g \
    --name ${CONTAINER_NAME} \
    ${image_name} \
    /bin/bash

# 检查容器是否启动成功
if [ $? -eq 0 ]; then
    echo "容器启动成功,容器名:${CONTAINER_NAME}"
else
    echo "容器启动失败!"
    exit 1
fi
# exit 0
# 步骤6: 将命令写入 ~/.bashrc 文件并执行
echo "步骤6: 将命令写入 ~/.bashrc 文件并执行..."

# 定义需要写入的命令列表(直接使用宿主机变量)
commands_to_write=(
    "cd /usr/local/Ascend/mindie/latest"
    "chmod 750 mindie-service"
    "chmod -R 550 mindie-service/bin"
    "chmod -R 500 mindie-service/bin/mindie_llm_backend_connector"
    "chmod 550 mindie-service/lib"
    "chmod 440 mindie-service/lib/*"
    "chmod 550 mindie-service/lib/grpc"
    "chmod 440 mindie-service/lib/grpc/*"
    "chmod -R 550 mindie-service/include"
    "chmod -R 550 mindie-service/scripts"
    "chmod 750 mindie-service/logs"
    "chmod 750 mindie-service/conf"
    "chmod 640 mindie-service/conf/config.json"
    "chmod 700 mindie-service/security"
    "chmod -R 700 mindie-service/security/*"
    "source /usr/local/Ascend/ascend-toolkit/set_env.sh"
    "source /usr/local/Ascend/nnal/atb/set_env.sh"
    "source /usr/local/Ascend/atb-models/set_env.sh"
    "unset HCCL_OP_EXPANSION_MODE"
    "export ATB_LLM_HCCL_ENABLE=1"
    "export ATB_LLM_COMM_BACKEND=\"hccl\""
    "export PYTORCH_NPU_ALLOC_CONF=\"expandable_segments:True\""
    "export MIES_CONTAINER_IP=${ip_addr_now}"
    "export RANKTABLEFILE=$mount_path/rank_table_file.json"
    "export HCCL_DETERMINISTIC=false"
    "export PARALLEL_PARAMS=\"[1,16,1,16,-1,-1]\""
    "export ATB_WORKSPACE_MEM_ALLOC_ALG_TYPE=3"
    "export ATB_WORKSPACE_MEM_ALLOC_GLOBAL=1"
    "export NPU_MEMORY_FRACTION=0.96"
    "export HCCL_CONNECT_TIMEOUT=7200"
    "export HCCL_EXEC_TIMEOUT=0"
    # "export MINDIE_LOG_TO_STDOUT=1"
)

# 将命令列表转换为单行字符串(避免引号嵌套问题)
commands_str=$(printf "%s\n" "${commands_to_write[@]}")

# 在容器内执行命令(使用单引号包裹)
docker exec ${CONTAINER_NAME} bash -c '
    # 逐行处理命令
    while IFS= read -r cmd; do
        # 检查命令是否已存在
        if ! grep -Fxq "$cmd" ~/.bashrc; then
            echo "$cmd" >> ~/.bashrc
        fi
    done <<< "$1"

    # 使配置生效
    # source ~/.bashrc
' _ "${commands_str}"  # 通过参数传递命令列表

# 检查命令是否执行成功
if [ $? -eq 0 ]; then
    echo "命令已成功写入 ~/.bashrc 并执行。"
else
    echo "命令写入或执行失败!"
    echo "删除容器${CONTAINER_NAME}"
    docker stop ${CONTAINER_NAME} && docker rm ${CONTAINER_NAME}
    echo "删除成功"
    exit 1
fi


# 容器内执行命令
echo "环境变量设置的当前节点ip为:"
docker exec ${CONTAINER_NAME} bash -c 'source ~/.bashrc && echo $MIES_CONTAINER_IP'
docker exec ${CONTAINER_NAME} bash -c "cat $RANK_TABLE_FILE"


docker exec ${CONTAINER_NAME} bash -c "
    cd /usr/local/Ascend/mindie/latest/mindie-service/conf && \
    sed -i 's|\"ipAddress\" : \".*\"|\"ipAddress\" : \"${node_ips[0]}\"|g' config.json && \
    sed -i 's|\"managementIpAddress\" : \".*\"|\"managementIpAddress\" : \"${node_ips[0]}\"|g' config.json && \
    sed -i 's|\"modelName\" : \".*\"|\"modelName\" : \"${model_name}\"|g' config.json && \
    sed -i 's|\"modelWeightPath\" : \".*\"|\"modelWeightPath\" : \"${model_path}\"|g' config.json && \
    sed -i 's|\"maxPrefillBatchSize\" : .*|\"maxPrefillBatchSize\" : ${maxPrefillBatchSize},|g' config.json && \
    sed -i 's|\"maxSeqLen\" : .*|\"maxSeqLen\" : ${maxSeqLen},|g' config.json && \
    sed -i 's|\"maxInputTokenLen\" : .*|\"maxInputTokenLen\" : ${maxInputTokenLen},|g' config.json && \
    sed -i 's|\"maxPrefillTokens\" : .*|\"maxPrefillTokens\" : ${maxPrefillTokens},|g' config.json && \
    sed -i 's|\"httpsEnabled\" : .*|\"httpsEnabled\" : ${httpsEnabled},|g' config.json && \
    sed -i 's|\"multiNodesInferEnabled\" : .*|\"multiNodesInferEnabled\" : ${multiNodesInferEnabled},|g' config.json && \
    sed -i 's|\"interCommTLSEnabled\" : .*|\"interCommTLSEnabled\" : ${interCommTLSEnabled},|g' config.json && \
    sed -i 's|\"interNodeTLSEnabled\" : .*|\"interNodeTLSEnabled\" : ${interNodeTLSEnabled},|g' config.json && \
    sed -i 's|\"maxIterTimes\" : .*|\"maxIterTimes\" : ${maxIterTimes},|g' config.json
"

# 检查修改后的结果
echo "修改后的值如下所示:"
docker exec ${CONTAINER_NAME} bash -c "cd /usr/local/Ascend/mindie/latest/mindie-service/conf && 
    cat config.json"



docker exec ${CONTAINER_NAME} bash -c "source ~/.bashrc && rm -rf /root/mindie/cache/* && cd /usr/local/Ascend/mindie/latest/mindie-service && 
    nohup ./bin/mindieservice_daemon &> $mount_path/start_log_$ip_addr_now.txt &"



# 查看容器日志
echo "查看容器日志:"
docker logs -f ${CONTAINER_NAME}

echo "所有步骤执行完毕。"
提示

您即将访问非华为云网站,请注意账号财产安全

文档反馈

文档反馈

意见反馈

0/500

标记内容

同时提交标记内容