文档HOT
下载
中文
注册

环境准备

环境搭建

进行算子开发之前，需要安装驱动固件和CANN软件包，完成开发环境和运行环境的准备，驱动固件的安装步骤可参见《CANN软件安装指南》的“安装驱动和固件”章节，本节不再给出安装示例。

若要使能msdebug工具，安装驱动时需要与--debug参数配合使用。

下面以开发环境和运行环境合设、操作系统架构为X86_64、root用户操作为例，介绍CANN软件安装的步骤，其他场景的安装步骤类似，详细的可参见《CANN软件安装指南》。

  1. 安装第三方依赖。
    • Ubuntu系统（Debian、UOS20、Linux等系统操作一致）： 
      apt-get install -y gcc g++ make cmake zlib1g zlib1g-dev openssl libsqlite3-dev libssl-dev libffi-dev unzip pciutils net-tools libblas-dev gfortran libblas3
    • openEuler系统（EulerOS、CentOS、BCLinux等系统操作一致）： 
      yum install -y gcc gcc-c++ make cmake unzip zlib-devel libffi-devel openssl-devel pciutils net-tools sqlite-devel lapack-devel gcc-gfortran
  2. 安装Python。

    可执行如下命令检查系统是否安装满足要求的python开发环境（要求python3.7.5~3.7.11、python3.8.0~3.8.11、python3.9.0~3.9.7和python3.10.0~3.10.12）。

    python3 --version
pip3 --version
    如果返回信息满足Python版本要求，则直接进入下一步，若不满足请参考《CANN软件安装指南》中的“安装开发环境 > 安装依赖”根据对应的操作系统安装python。

    若您后续需要安装PyTorch 2.1.0版本或者PyTorch 2.2.0的AI框架，您的python版本需要为python3.8.0~3.8.11、python3.9.0~3.9.7或python3.10.0~3.10.12，python3.7.5~3.7.11环境仅支持安装PyTorch 1.11.0。

  3. 安装相关依赖。
    pip3 install attrs numpy decorator sympy cffi pyyaml pathlib2 psutil protobuf scipy requests absl-py wheel typing_extensions
  4. 安装CANN开发套件包。
    1. 将CANN开发套件包上传至安装环境的任意目录，执行如下命令增加对软件包的可执行权限。 
      chmod +x Ascend-cann-toolkit_8.0.RC2.alpha002_linux-x86_64.run
    2. 执行如下命令校验软件包的一致性和完整性。 
      ./Ascend-cann-toolkit_8.0.RC2.alpha002_linux-x86_64.run --check
    3. 执行如下命令安装CANN开发套件包。 
      ./Ascend-cann-toolkit_8.0.RC2.alpha002_linux-x86_64.run --install

      用户需签署华为企业业务最终用户许可协议（EULA）后进入安装流程，根据回显页面输入y或Y接受协议，输入其他任意字符为拒绝协议，确认接受协议后开始安装。安装完成后，若显示如下信息，则说明软件安装成功。

      [INFO] Ascend-cann-toolkit install success
    4. 配置CANN环境变量。 
      source /usr/local/Ascend/ascend-toolkit/set_env.sh

设置环境变量

  • 设置公共环境变量

    安装CANN软件后，使用CANN运行用户编译、运行时，需要以CANN运行用户登录环境，执行. ${install_path}/set_env.sh命令设置环境变量。其中${install_path}为CANN软件的安装目录。

  • 设置Python相关环境变量
    存在多个Python3版本时，以指定python3.7.5为例，请根据实际修改。 
    export PATH=/usr/local/python3.7.5/bin:$PATH
#设置python3.7.5库文件路径
export LD_LIBRARY_PATH=/usr/local/python3.7.5/lib:$LD_LIBRARY_PATH
  • 设置AscendCL应用编译所需其他环境变量
    ${INSTALL_DIR}请替换为CANN软件安装后文件存储路径。例如，若安装的Ascend-cann-toolkit软件包，则安装后文件存储路径为：$HOME/Ascend/ascend-toolkit/latest。 
    export DDK_PATH=${INSTALL_DIR}
export NPU_HOST_LIB=${INSTALL_DIR}/{arch-os}/devlib

上述环境变量只在当前窗口生效，用户可以将上述命令写入~/.bashrc文件，使其永久生效，操作如下：

  1. 以安装用户在任意目录下执行vi ~/.bashrc，在该文件最后添加上述内容。
  2. 执行:wq!命令，保存文件并退出。
  3. 执行source ~/.bashrc命令，使环境变量生效。
父主题： 算子开发工具