昇腾社区首页
中文
注册

开发前准备

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

安装驱动固件与CANN软件包

准备带有昇腾AI处理器的硬件环境,并安装驱动固件和CANN软件包,具体安装步骤请参见CANN 软件安装指南。安装完成后:

  • ${INSTALL_DIR}/opp/built-in/op_proto/inc”下提供了CANN算子原型定义,用于通过算子原型构建Graph。
  • ${INSTALL_DIR}/include/graph”下提供了Graph构建接口。
  • ${INSTALL_DIR}/include/ge”下提供了Graph运行接口。
  • ${INSTALL_DIR}/<arch>-linux/devlib”下为相关依赖库。<arch>为操作系统架构。

${INSTALL_DIR}请替换为CANN软件安装后文件存储路径。若安装的Ascend-cann-toolkit软件包,以root安装举例,则安装后文件存储路径为:/usr/local/Ascend/ascend-toolkit/latest。

设置环境变量

  • 必选环境变量

    安装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处理器是否支持,当前支持的算子请参见CANN算子规格说明。如果不支持或不满足实际需要,可参见如下方式自定义算子:

      参见Ascend C算子开发指南自定义Ascend C算子,并将算子部署到硬件环境。

  • 如果是将原始模型解析为Graph,需要确认原始网络中的算子在昇腾AI处理器是否支持,当前支持的算子请参见CANN算子规格说明。如果不支持或不满足实际需要,可参见如下方式自定义算子:

    参见Ascend C算子开发指南自定义Ascend C算子,并将算子部署到硬件环境。