下载
中文
注册

环境准备

进行Ascend C算子开发之前,需要安装驱动固件CANN软件包,完成环境准备。

驱动固件(仅昇腾设备需要)的安装步骤可参见CANN软件安装指南的“安装NPU驱动固件”章节,本节不再给出安装示例。

本节以训练&推理&开发调试业务场景、操作系统架构X86_64、root用户操作为例,介绍CANN软件包安装的步骤,其他场景的安装步骤类似,详细的可参见CANN软件安装指南

  1. 安装相关依赖。
    • Debian系列(Ubuntu、Debian、UOS20、UOS20 SP1):
      1
      apt-get install -y gcc make net-tools python3 python3-dev python3-pip
      
    • openEuler系列(openEuler、CentOS、Kylin、BCLinux、BC-Linux-for-Euler、UOS201050e、UOS20 1020e、UOSV20、AntOS、CTyunOS、CULinux、Tlinux):
      1
      yum install -y gcc make net-tools python3 python3-devel python3-pip
      
  2. 检查系统是否安装满足版本要求的Python开发环境(支持Python3.7.x至3.11.4版本)。
    1
    2
    python3 --version
    pip3 --version
    

    如果返回信息满足Python版本要求,则直接进入下一步,若不满足请参考CANN软件安装指南中的“编译安装Python3.7.5”章节。

  3. 执行如下命令安装所需的Python第三方库。
    1
    pip3 install attrs cython numpy==1.24.0 decorator sympy cffi pyyaml pathlib2 psutil protobuf==3.20 scipy requests absl-py
    

    当前命令会安装最新或指定版本的依赖,若安装报错或更多版本要求请参考CANN软件安装指南的“依赖列表”。

  4. 安装CANN开发套件包。
    1. 将CANN开发套件包上传至安装环境的任意目录,执行如下命令增加对软件包的可执行权限。
      chmod +x Ascend-cann-toolkit_8.0.RC3.alpha001_linux-x86_64.run
    2. 执行如下命令校验软件包的一致性和完整性。
      ./Ascend-cann-toolkit_8.0.RC3.alpha001_linux-x86_64.run --check
    3. 执行如下命令安装CANN开发套件包。
      ./Ascend-cann-toolkit_8.0.RC3.alpha001_linux-x86_64.run --install

      用户需签署华为企业业务最终用户许可协议(EULA)后进入安装流程,根据回显页面输入y或Y接受协议,安装完成后,若显示如下信息,则说明软件安装成功。

      1
      [INFO] Ascend-cann-toolkit install success
      
    4. 配置CANN环境变量。
      1
      source /usr/local/Ascend/ascend-toolkit/set_env.sh
      
  • 对于Ascend C算子的开发,并非必须安装驱动固件。对于非昇腾设备,可以利用CPU仿真环境先行进行算子开发和测试,并在准备就绪后,利用昇腾设备进行加速计算。非昇腾设备的安装请参考CANN软件安装指南中“附录B:常用操作 > 在非昇腾设备上安装CANN”章节。
  • 安装第三方依赖时,gcc/g++的版本应满足CANN软件安装指南中所列出的要求。
  • 使用Ascend C算子工程(kernel直调工程和自定义算子工程)要求安装cmake(版本为3.16及以上),如下示例以安装3.16.0版本的cmake(linux系统、x86_64架构)为例。
    mkdir -p cmake-3.16 && wget -qO- "https://cmake.org/files/v3.16/cmake-3.16.0-linux-x86_64.tar.gz" | tar --strip-components=1 -xz -C cmake-3.16
    export PATH=`pwd`/cmake-3.16/bin:$PATH