下载
中文
注册

环境准备

进行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 cmake 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 cmake python3 python3-devel python3-pip
      
    • 检查依赖版本。
      • Python:要求Python3.7.x至3.11.4版本。
      • pip3:需与已安装的Python版本配套。
      • cmake:要求3.16及以上版本。
      • gcc:要求7.3.0及以上版本。

      运行以下命令检查版本:

      1
      2
      3
      4
      python3 --version
      pip3 --version
      cmake --version
      gcc --version
      

      如果版本不符合要求:

      • 对于Python、pip3、gcc,请参考CANN 软件安装指南中“附录B:常用操作”章节安装满足要求的版本。
      • 对于cmake,请参考cmake安装进行升级。
  2. 执行如下命令安装所需的Python第三方库。
    1
    pip3 install attrs cython numpy==1.24.0 decorator sympy cffi pyyaml pathlib2 psutil protobuf==3.20 scipy requests absl-py
    

    当前命令会安装最新或指定版本的依赖,如果安装过程中报错,或需要了解更详细的版本要求,请参考CANN 软件安装指南的“依赖列表”。

  3. 安装CANN开发套件包。
    1. 将CANN开发套件包上传至安装环境的任意目录,执行如下命令增加对软件包的可执行权限。
      chmod +x Ascend-cann-toolkit_XXX_linux-x86_64.run
    2. 执行如下命令校验软件包的一致性和完整性。
      ./Ascend-cann-toolkit_XXX_linux-x86_64.run --check
    3. 执行如下命令安装CANN开发套件包。
      ./Ascend-cann-toolkit_XXX_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”章节。
  • 使用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