拓冰建站拓冰建站
首页 / 资讯中心 / 正文

CANN/ge集群配置参数指南

--cluster_config【免费下载链接】geGEGraph Engine是面向昇腾的图编译器和执行器提供了计算图优化、多流并行、内存复用和模型下沉等技术手段加速模型执行效率减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge产品支持情况Ascend 950PR/Ascend 950DT不支持Atlas A3 训练系列产品/Atlas A3 推理系列产品不支持Atlas A2 训练系列产品/Atlas A2 推理系列产品不支持Atlas 200I/500 A2 推理产品不支持Atlas 推理系列产品不支持Atlas 训练系列产品不支持IPV350不支持功能说明设置集群配置文件路径用于配置目标执行逻辑设备信息以生成HCCL任务多卡分布式训练场景下使用。若模型中包含通信算子当前仅支持AllGather通信算子该参数必填。关联参数无。参数取值参数值逻辑拓扑文件的路径和文件名。参数值格式路径和文件名支持大小写字母a-zA-Z、数字0-9、下划线_、短横线-、句点.、中文汉字。参数值约束配置文件中的内容必须为JSON格式。推荐配置及收益无。示例将配置文件文件名为举例为cluster_config.json上传到ATC工具所在服务器例如上传到_$HOME/conf_使用示例如下atc --modelxxx.air --framework1 --soc_versionsoc_version --output$HOME/out --cluster_config$HOME/conf/cluster_config.json ...逻辑拓扑文件示例如下{ RankTable: { status: completed, version: 1.2, server_count: 2, server_list: [ { server_id: node_0, device: [ { device_id: 0, rank_id: 0 } ] }, { server_id: node_1, device: [ { device_id: 0, rank_id: 1 } ] } ] }, HcclCommConfig: { hcclCommName: global name } }参数说明解释如下RankTable配置的核心定义了参与计算的节点和rank信息。status必选rank table可用标识。completed表示rank table可用。initializing表示rank table不可用。version必选rank table模板版本信息请配置为1.2。server_count可选参与集合通信的AI Server个数。server_list必选参与集合通信的AI Server列表。server_id必选AI Server标识字符串类型长度小于等于64请确保全局唯一。配置示例node_0。device必选Device列表。device_id必选AI处理器的物理ID即Device在AI Server上的序列号。rank_id必选rank唯一标识请配置为整数从0开始配置且全局唯一取值范围[0, 总Device数量-1]。HcclCommConfigHCCL通信层的全局配置信息。hcclCommName必选全局通信组的名称或标识符。依赖约束无。【免费下载链接】geGEGraph Engine是面向昇腾的图编译器和执行器提供了计算图优化、多流并行、内存复用和模型下沉等技术手段加速模型执行效率减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
分享:

看完干货,该让你的企业上线了

免费需求沟通 · 48 小时内出具建站方案 · 河南本地可上门