下载
中文
注册

开发前准备

构图之前,需要先搭建构图环境,包括安装软件包以及设置环境等,本章节给出详细介绍。

安装软件包

参见CANN 软件安装指南安装CANN软件包。安装完成后:

  • “Ascend-cann-toolkit安装目录/ascend-toolkit/latest/opp/built-in/op_proto/inc”下提供了AI Core和AI CPU算子原型定义,用于通过算子原型构建Graph。
  • “Ascend-cann-toolkit安装目录/ascend-toolkit/latest/include/graph”下提供了Graph构建接口。
  • “Ascend-cann-toolkit安装目录/ascend-toolkit/latest/include/ge”下提供了Graph运行接口。
  • “Ascend-cann-toolkit安装目录/ascend-toolkit/latest/compiler/lib64/stub”下为相关依赖库。

设置环境变量

  • 必选环境变量

    安装CANN软件后,使用CANN运行用户进行编译、运行时,需要以CANN运行用户登录环境,执行source ${install_path}/set_env.sh命令设置环境变量。其中${install_path}为CANN软件的安装目录,例如:/usr/local/Ascend/ascend-toolkit。

  • 可选环境变量
    若开发者期望程序编译运行过程中产生的文件落盘到归一目录,可通过环境变量ASCEND_CACHE_PATHASCEND_WORK_PATH分别设置共享文件的存储路径与进程独享文件的存储路径。
    export ASCEND_CACHE_PATH=/repo/task001/cache
    export ASCEND_WORK_PATH=/repo/task001/172.16.1.12_01_03

    配置此环境变量前,请使用env命令查询ASCEND_CACHE_PATHASCEND_WORK_PATH环境变量是否已存在,建议系统各功能使用统一的规划路径。

网络分析

  • 如果通过算子原型构建Graph,需要根据原始网络,明确如下信息:
    1. 网络中包含哪些算子,以及这些算子的输入、输出、属性等信息。
    2. 网络中算子之间的关联关系。
    3. 确认原始网络中的算子在昇腾AI处理器是否支持,当前支持的算子请参考算子加速库接口参考。如果不支持或不满足实际需要,请参考TBE&AI CPU自定义算子开发指南自定义算子,并将算子部署到硬件环境。
  • 如果是将原始模型解析为Graph,需要确认原始网络中的算子在昇腾AI处理器是否支持,当前支持的算子请参考算子加速库接口参考。如果不支持或不满足实际需要,请参考TBE&AI CPU自定义算子开发指南自定义算子,并将算子部署到硬件环境。