安装CANN(Pip在线安装)
前提条件
安装前需确保已具备Python环境及pip3,当前CANN支持Python3.7.x至3.13.x版本,若不满足可执行以下命令安装。
1 | sudo apt-get install -y python3 python3-pip |
1 | sudo yum install -y python3 python3-pip |
若安装失败、版本不满足或未包含动态库libpython3.x.so,请参考编译安装Python操作。
配置昇腾源
执行如下命令配置昇腾源,否则将无法安装。
pip config set global.extra-index-url https://ascend.devcloud.huaweicloud.com/cann/pypi/simple pip config set global.trusted-host ascend.devcloud.huaweicloud.com
安装Toolkit开发套件包
Pip方式安装的CANN Toolkit开发套件包,主要用于框架侧运行训练和推理业务。若需要开发调试场景下(算子/应用/模型的开发和编译等)的Toolkit包,请选择其他安装方式。
安装Toolkit软件前请确保安装目录可用空间大于10G,如不满足请清理空间或更换安装目录。
- 执行安装命令。
1pip install ascend-cann-toolkit==9.0.0
软件包会安装在“<Python所在路径>/site-packages/ascend”目录下,可以执行pip show ascend-cann-toolkit查询软件包具体路径。
- 配置环境变量,以安装路径在“${HOME}/lib/python3.12/site-packages/ascend”下为例。
1source ${HOME}/lib/python3.12/site-packages/ascend/cann-9.0.0/set_env.sh
上述环境变量配置只在当前窗口生效,用户可以按需将以上命令写入环境变量配置文件(如.bashrc文件)。
安装ops算子包
CANN算子包集成一系列库文件,包括算子基础框架,算子库(包含math、nn、cv和transformer)、TBE算子、HCCL集合通信库、HIXL单边通信库和 DVPP库,具体文件包括单算子API执行(如aclnn类API)动态库/静态库文件、算子源码以及kernel二进制文件,优化高性能计算场景下的整体运行能力。
安装ops算子包前,需已安装兼容版本的Toolkit软件包,并安装在同一路径下,请选择运行设备对应的ops软件包。多个芯片的ops算子包暂不支持安装在同一路径下,用户可以在不同路径下分别安装Toolkit和不同芯片的ops,以满足多芯片环境下的开发和部署需求。
产品类型 |
安装命令 |
||
|---|---|---|---|
Atlas 350 加速卡 |
|
||
Atlas A3 系列产品 |
|
||
Atlas A2 系列产品 |
|
||
Atlas 训练系列产品 |
|
||
Atlas 推理系列产品 |
|
||
Atlas 200I/500 A2 推理产品 |
|