Installing CANN (Online Installation Using pip)
Prerequisites
Before installation, ensure that the Python environment and pip3 are available. CANN currently supports Python versions 3.7.x to 3.13.x. If this requirement is not met, run the following commands to install them:
1 | sudo apt-get install -y python3 python3-pip |
1 | sudo yum install -y python3 python3-pip |
If the installation fails, the version does not meet the requirements, or the dynamic library libpython3.x.so is not included, perform the operations in Compiling and Installing Python.
Configuring the Ascend Repository
Configure the Ascend repository. Otherwise, the installation will fail.
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
Installing the Toolkit
The pip-installed CANN Toolkit is intended only for framework-side training and inference workloads. If you need to install the Toolkit for development and debugging (such as development and compilation of operators, applications, and models), select another installation method.
Before installing Toolkit, ensure that the available space of the installation directory is greater than 10 GB. Otherwise, clear the space or change the installation directory.
- Run the installation command:
1pip install ascend-cann-toolkit==9.0.0
The software package is installed in the <Python path>/site-packages/ascend directory. You can run the pip show ascend-cann-toolkit command to query the specific path of the software package.
- Set the environment variable. The following uses installation path ${HOME}/lib/python3.12/site-packages/ascend as an example.
1source ${HOME}/lib/python3.12/site-packages/ascend/cann-9.0.0/set_env.sh
The preceding environment variables take effect only in the current window. You can add the preceding command to the environment variable configuration file (for example, the .bashrc file) as required.
Installing the ops
The CANN ops package integrates a comprehensive suite of libraries, including the operator foundational framework, operator libraries (covering math, nn, cv, and transformer), TBE operators, HCCL, HIXL, and the DVPP library. Specifically, it provides dynamic and static library files for single-operator API execution (e.g., aclnn-class APIs), operator source code, and kernel binaries, ultimately optimizing overall execution performance in high-performance computing (HPC) scenarios.
Before installing the ops package, ensure that the Toolkit software package of a compatible version has been installed in the same path. Select the ops software package corresponding to the running device. Currently, installing ops packages for multiple chips under the same path is not supported. You can install the Toolkit and the corresponding ops packages for different chips in separate paths to meet development and deployment requirements in multi-chip environments.
Product Type |
Installation Command |
||
|---|---|---|---|
Atlas 350 Accelerator Card |
|
||
Atlas A3 series |
|
||
Atlas A2 series |
|
||
Atlas training series |
|
||
Atlas inference series |
|
||
Atlas 200I/500 A2 inference products |
|