Pyatc接口【免费下载链接】geGEGraph Engine是面向昇腾的图编译器和执行器提供了计算图优化、多流并行、内存复用和模型下沉等技术手段加速模型执行效率减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge产品支持情况全量芯片支持。功能说明pyatc是ATCAscend Tensor Compiler离线模型编译工具的Python封装接口在功能上与ATC命令行工具完全等价旨在为Python用户提供更灵活的集成体验。pyatc与ATC命令行工具的区别是ATC命令行工具作为独立子进程启动编译过程所使用的Python解释器由atc进程自行解析可能与用户当前终端环境如python3不一致容易引发依赖冲突或路径问题。pyatc直接在当前主进程中执行编译逻辑复用用户当前的Python解释器及环境变量确保编译环境与运行环境完全一致避免了进程间的环境隔离问题。调用方式先参见环境准备完成相关环境变量的设置然后根据实际使用场景选择以下任一入口使用命令行入口pyatc [参数]使用Python模块入口python3 -m ge.pyatc [参数]参数说明pyatc的参数与ATC命令行工具完全一致详细功能介绍请参见《ATC离线模型编译工具》。返回值说明调用结束后通过进程退出码exit code表示执行结果0执行成功。非 0执行失败错误信息输出至标准输出。调用示例pyatc --modelresnet50.onnx --framework5 --soc_versionsoc_version --outputresnet50pyatc --modelresnet50.onnx --framework5 --soc_versionMC62CM12A* --outputresnet50pyatc --modelresnet50.onnx --framework5 --soc_versionAscend035 --outputresnet50_soc_version_取值或者查询方法如下针对如下产品在安装AI处理器的服务器执行npu-smi info命令进行查询获取Name信息。实际配置值为AscendName例如Name取值为_xxxyy_实际配置值为Ascend_xxxyy_。Atlas A2 训练系列产品/Atlas A2 推理系列产品Atlas 200I/500 A2 推理产品Atlas 推理系列产品Atlas 训练系列产品针对Atlas A3 训练系列产品/Atlas A3 推理系列产品在安装AI处理器的服务器执行npu-smi info -t board -iid-c_chip_id_命令进行查询获取Chip Name和NPU Name信息实际配置值为Chip Name_NPU Name。例如Chip Name取值为Ascend_xxx_NPU Name取值为1234实际配置值为Ascend_xxx____1234。其中id设备id通过npu-smi info -l命令查出的NPU ID即为设备id。chip_id芯片id通过npu-smi info -m命令查出的Chip ID即为芯片id。针对Ascend 950PR/Ascend 950DT在安装AI处理器的服务器执行npu-smi info -t board -i_id_命令进行查询获取Chip Name和NPU Name信息实际配置值为Chip Name_NPU Name。例如Chip Name取值为Ascend_xxx_NPU Name取值为1234实际配置值为Ascend_xxx____1234。其中id为设备id通过npu-smi info -l命令查出的NPU ID即为设备id。BS9SX1A AI处理器参数值BS9SX1A*BS9SX2A AI处理器参数值BS9SX2A*其中*可能根据芯片性能提升等级、芯片核数使用等级等因素会有不同的取值例如取值可以为A、B或C请根据实际情况获取对应的取值。【免费下载链接】geGEGraph Engine是面向昇腾的图编译器和执行器提供了计算图优化、多流并行、内存复用和模型下沉等技术手段加速模型执行效率减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考