昇腾社区首页
中文
注册
开发者
下载
支持
积分兑换 NEW

安装CANN

本节将介绍通过Yum快速安装CANN软件的操作步骤。

安装依赖

安装前需确保已具备Python环境及pip3，当前CANN支持的Python版本为Python3.7.x至3.11.4，若不满足可执行以下命令安装。
1
sudo yum install -y python3 python3-pip

若安装失败、版本不满足或者未包含动态库libpython3.x.so请参考编译安装Python操作。

配置昇腾源

执行如下命令配置昇腾源，否则将无法安装。

sudo curl https://repo.oepkgs.net/ascend/cann/ascend.repo -o /etc/yum.repos.d/ascend.repo && yum makecache

安装Toolkit开发套件包

CANN Toolkit开发套件包，在训练&推理&开发调试场景下安装，主要用于训练和推理业务、模型转换、算子/应用/模型的开发和编译。

安装Toolkit软件前请确保安装目录可用空间大于10G，如不满足请清理空间或更换安装目录。

  1. 执行安装命令。
    1
    sudo yum install -y Ascend-cann-toolkit

    Yum在线安装时仅支持安装在默认路径“/usr/local/Ascend”下。

  2. 配置环境变量。
    1
    source /usr/local/Ascend/ascend-toolkit/set_env.sh

    上述环境变量配置只在当前窗口生效，用户可以按需将以上命令写入环境变量配置文件（如.bashrc文件）。

安装NNAE深度学习引擎包

CANN NNAE深度学习引擎包，在训练&推理场景下安装，主要用于训练和推理业务。

  1. 执行安装命令。
    1
    sudo yum install -y Ascend-cann-nnae

    Yum在线安装时仅支持安装在默认路径“/usr/local/Ascend”下。

  2. 配置环境变量。
    1
    source /usr/local/Ascend/nnae/set_env.sh

    上述环境变量配置只在当前窗口生效，用户可以按需将以上命令写入环境变量配置文件（如.bashrc文件）。

安装NNRT离线推理引擎包

CANN NNRT离线推理引擎包，在边缘推理场景下安装，仅支持离线推理，主要用于应用程序的模型推理。

  1. 执行安装命令。
    1
    sudo yum install -y Ascend-cann-nnrt

    Yum在线安装时仅支持安装在默认路径“/usr/local/Ascend”下。

  2. 配置环境变量。
    1
    source /usr/local/Ascend/nnrt/set_env.sh

    上述环境变量配置只在当前窗口生效，用户可以按需将以上命令写入环境变量配置文件（如.bashrc文件）。

安装Kernels算子包

CANN二进制算子包，包括单算子API执行（例如aclnn类API）动态库/静态库文件，以及kernel二进制文件。

Kernels算子包安装之前，需已安装配套版本的Toolkit并配置环境变量。

Kernels算子包安装之前，需已安装配套版本的NNAE并配置环境变量。

Kernels算子包安装之前，需已安装配套版本的NNRT并配置环境变量。

表1 安装Kernels算子包

产品类型

安装命令

Atlas A3 系列产品

 
1
sudo yum install -y Atlas-A3-cann-kernels-8.3.RC1

Atlas A2 系列产品

 
1
sudo yum install -y Ascend-cann-kernels-910b-8.3.RC1

Atlas 训练系列产品

 
1
sudo yum install -y Ascend-cann-kernels-910-8.3.RC1

Atlas 推理系列产品

 
1
sudo yum install -y Ascend-cann-kernels-310p-8.3.RC1

Atlas 200I/500 A2 推理产品

 
1
sudo yum install -y Ascend-cann-kernels-310b-8.3.RC1

安装NNAL神经网络加速库（可选）

NNAL神经网络加速库中提供了ATB（Ascend Transformer Boost）加速库和SiP（AscendSiPBoost）信号处理加速库。

加速库安装之前，需已安装同一版本的Toolkit并配置环境变量。

加速库安装之前，需已安装同一版本的NNAE并配置环境变量。

  1. 执行安装命令。
    1
    sudo yum install -y Ascend-cann-nnal

    Yum在线安装时仅支持安装在默认路径“/usr/local/Ascend”下。

  2. 配置环境变量。
    需注意，不支持同时配置ATB和SiP加速库的环境变量脚本。
    • ATB加速库：
      1
      source /usr/local/Ascend/nnal/atb/set_env.sh
    • SiP加速库：
      1
      source /usr/local/Ascend/nnal/asdsip/set_env.sh

    上述环境变量配置只在当前窗口生效，用户可以按需将以上命令写入环境变量配置文件（如.bashrc文件）。

父主题： 安装指南