昇腾社区首页
中文
注册
开发者
下载

环境准备

使用图模式进行开发之前,需要先搭建相应环境,包括安装软件包以及设置环境变量等,本章节给出详细介绍。

安装驱动固件与CANN软件包

准备带有昇腾AI处理器的硬件环境,参见CANN 软件安装指南完成环境搭建,并确保已安装CANN Toolkit开发套件包和ops算子包(针对8.5.0及之后版本,编译Graph为离线模型时,必须安装与目标昇腾AI处理器相匹配的ops算子包,否则会导致编译失败)。安装完成后:

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

其中,${INSTALL_DIR}请替换为CANN软件安装后文件存储路径。以root用户安装为例,则安装后文件存储路径为:/usr/local/Ascend/cann。

设置环境变量

  • 必选环境变量
    安装CANN软件后,使用CANN运行用户进行编译、运行时,需要以CANN运行用户登录环境,执行如下环境变量:
    source /usr/local/Ascend/cann/set_env.sh

    其中,/usr/local/Ascend/为root用户的默认安装路径,请根据实际情况进行替换。

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