开发者
资源

安装CANN(Pip在线安装)

前提条件

安装前需确保已具备Python环境及pip3,当前CANN支持Python3.7.x至3.13.x版本,若不满足可执行以下命令安装。

Debian、Ubuntu、veLinux
1
sudo apt-get install -y python3 python3-pip
openEuler、CentOS、Kylin、BCLinux、UOS、AntOS、CTyunOS、CULinux、Tlinux、MTOS、vesselOS
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,如不满足请清理空间或更换安装目录。
  • 必须同时安装Toolkitops
  1. 执行安装命令。
    1
    pip install ascend-cann-toolkit==9.0.0b2
    

    pip在线安装时会安装在“<python_path>/site-packages/ascend”目录下,<python_path>为Python所在安装路径,可以执行python3 -V查询。

  2. 配置环境变量,以安装路径在${HOME}/lib/python3.12下为例。
    1
    source ${HOME}/lib/python3.12/site-packages/ascend/cann-9.0.0-beta.2/set_env.sh
    

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

安装ops算子包

ops算子包安装之前,需已安装同一版本的Toolkit,并安装在同一路径下。多个芯片的ops算子包暂不支持安装在同一路径下,用户可以在不同路径下分别安装Toolkit和不同芯片的ops,以满足多芯片环境下的开发和部署需求。

表1 安装ops算子包

产品类型

安装命令

Atlas A3 系列产品

1
pip install ascend-cann-a3-ops==9.0.0b2

Atlas A2 系列产品

1
pip install ascend-cann-910b-ops==9.0.0b2

Atlas 训练系列产品

1
pip install ascend-cann-910-ops==9.0.0b2

Atlas 推理系列产品

1
pip install ascend-cann-310p-ops==9.0.0b2

Atlas 200I/500 A2 推理产品

1
pip install ascend-cann-310b-ops==9.0.0b2