安装CANN
本节提供了通过Conda快速安装CANN软件的操作步骤,支持的CANN软件包及版本详见软件包列表。
前提条件
Conda虚拟环境的安装目录(及其所有上级目录)都需有755权限。
配置昇腾源
执行如下命令配置昇腾源,否则将无法安装。
1 | conda config --add channels https://repo.huaweicloud.com/ascend/repos/conda/ |
安装Toolkit开发套件包
CANN Toolkit开发套件包,在训练&推理&开发调试场景下安装,主要用于训练和推理业务、模型转换、算子/应用/模型的开发和编译。
- 安装Toolkit软件前请确保安装目录可用空间大于10G,如不满足请清理空间或更换安装目录。
- CANN 8.5.0版本必须同时安装Toolkit和ops。
- 执行安装命令。
- 配置环境变量,以安装在/home/miniconda3/Ascend为例,请根据实际情况替换路径。
1source /home/miniconda3/Ascend/cann/set_env.sh
上述环境变量配置只在当前窗口生效,用户可以按需将以上命令写入环境变量配置文件(如.bashrc文件)。
安装ops算子包
CANN算子包集成一系列库文件,包括算子基础框架,算子库(包含math、nn、cv和transformer)、TBE算子、HCCL集合通信库、HIXL单边通信库和 DVPP库,具体文件包括单算子API执行(如aclnn类API)动态库/静态库文件、算子源码以及kernel二进制文件,优化高性能计算场景下的整体运行能力。
ops算子包安装之前,需已安装同一版本的Toolkit,并安装在同一路径下。多个芯片的ops算子包暂不支持安装在同一路径下,用户可以在不同路径下分别安装Toolkit和不同芯片的ops,以满足多芯片环境下的开发和部署需求。
产品类型 |
安装命令 |
||
|---|---|---|---|
Atlas A3 系列产品 |
|
||
Atlas A2 系列产品 |
|
||
Atlas 训练系列产品 |
|
||
Atlas 推理系列产品 |
|
||
Atlas 200I/500 A2 推理产品 |
|
安装NNAL神经网络加速库(可选)
NNAL神经网络加速库中提供了ATB(Ascend Transformer Boost)加速库和SiP(AscendSiPBoost)信号处理加速库。
加速库安装之前,需已安装同一版本的Toolkit并配置环境变量。
- 执行安装命令。
- 配置环境变量,以安装在/home/miniconda3/Ascend为例,请根据实际情况替换路径。
需注意,不支持同时配置ATB和SiP加速库的环境变量脚本。
上述环境变量配置只在当前窗口生效,用户可以按需将以上命令写入环境变量配置文件(如.bashrc文件)。