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

安装CANN

本节提供了通过Conda快速安装CANN软件的操作步骤,支持的CANN软件包及版本详见软件包列表

前提条件

Conda虚拟环境的安装目录需有755权限。

配置昇腾源

执行如下命令配置昇腾源,否则将无法安装。
1
conda config --add channels https://repo.huaweicloud.com/ascend/repos/conda/ 

安装Toolkit开发套件包

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

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

  1. 执行安装命令。
    1
    conda install ascend::cann-toolkit
    

    以上指令默认安装最新版本的软件包,若需安装其他版本的软件包,可参考软件包列表执行对应指令。

    软件包的默认安装路径为所在Conda虚拟环境的Ascend目录下,例如/home/miniconda3/Ascend。

  2. 配置环境变量,以安装在/home/miniconda3/Ascend为例,请根据实际情况替换路径。
    1
    source /home/miniconda3/Ascend/ascend-toolkit/set_env.sh
    

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

  3. 安装后检查。执行如下命令查询CANN版本信息,查询结果与安装软件包的版本一致时,则验证安装成功。
    1
    conda list | grep cann
    

安装Kernels算子包

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

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

表1 安装Kernels算子包

产品类型

安装命令

Atlas A3 系列产品

1
conda install ascend::a3-cann-kernels

Atlas A2 系列产品

1
conda install ascend::cann-kernels-910b

Atlas 训练系列产品

1
conda install ascend::cann-kernels-910

Atlas 推理系列产品

1
conda install ascend::cann-kernels-310p

Atlas 200I/500 A2 推理产品

1
conda install ascend::cann-kernels-310b

以上指令默认安装最新版本的软件包,需安装指定版本时请自行在软件包后添加版本号,单击软件包列表查看对应指令。

安装NNAL神经网络加速库(可选)

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

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

  1. 执行安装命令。
    1
    conda install ascend::cann-nnal
    

    以上指令默认安装最新版本的软件包,若需安装其他版本的软件包,可参考软件包列表执行对应指令。

    软件包的默认安装路径为所在Conda虚拟环境的Ascend目录下,例如/home/miniconda3/Ascend。

  2. 配置环境变量,以安装在/home/miniconda3/Ascend为例,请根据实际情况替换路径。
    需注意,不支持同时配置ATB和SiP加速库的环境变量脚本。
    • ATB加速库:
      1
      source /home/miniconda3/Ascend/nnal/atb/set_env.sh
      
    • SiP加速库:
      1
      source /home/miniconda3/Ascend/nnal/asdsip/set_env.sh
      

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

  3. 安装后检查。执行如下命令查询CANN版本信息,查询结果与安装软件包的版本一致时,则验证安装成功。
    1
    conda list | grep cann