ascendai/cann repository overview

⁠Atlas CANN

CANN (Compute Architecture for Neural Networks) is a heterogeneous computing architecture launched by Huawei for AI scenarios. It supports multiple AI frameworks on the upper layer and serves AI processors and programming on the lower layer, acting as a key bridge and a key platform for improving the computing efficiency of Atlas AI processors.

⁠Image Download

CANN 9.2.0-beta.2 has been released. Please visit the Mirror Download⁠ page to obtain it.

⁠CANN Base Container Images && CANN Development Container Images

The CANN base images are built on Ubuntu and openEuler operating systems. They include the CANN toolkit (Toolkit development suite, ops operator package, NNAL acceleration library) and the Python environment.

CANN Development Container Images are built on CANN Base Container Images. In addition to the CANN toolkit suite and Python runtime environment, they come with extra OS utilities(such as zip, vim, tree, etc.), selected Python plugins(such as wheel, pyyaml, setuptools, etc.), and GoogleTest pre-installed.

⁠Supported Tags and Dockerfile Links

⁠Tag Naming Rules

CANN Tags follow this pattern:

<cann_version>-<chip_series>-<os>-<python_version>

⁠LATEST CANN 9.2.0-beta.2

For tags associated with historical versions, please refer to Supported Tags⁠.

⁠Quick Start

⁠Prerequisites

⁠Install Driver

An Atlas NPU driver compatible with the CANN version inside the container must be installed on the host. Please visit the CANN compatibility site⁠ for the correspondence between driver and CANN versions.

⁠Run a CANN Container

export CANN_REPO=quay.io/ascend/cann
export CANN_TAG=9.2.0-beta.2-a3-ubuntu22.04-py3.12

docker run \
    --name cann_container \
    --device /dev/davinci0 \
    --device /dev/davinci_manager \
    --device /dev/devmm_svm \
    --device /dev/hisi_hdc \
    -v /usr/local/dcmi:/usr/local/dcmi \
    -v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi \
    -v /usr/local/Ascend/driver/lib64/:/usr/local/Ascend/driver/lib64/ \
    -v /usr/local/Ascend/driver/version.info:/usr/local/Ascend/driver/version.info \
    -v /etc/ascend_install.info:/etc/ascend_install.info \
    -it $CANN_REPO:$CANN_TAG bash

⁠Run a CANN Container on 950 Series aarch64 Products

export CANN_REPO=quay.io/ascend/cann
export CANN_TAG=9.2.0-beta.2-950-openeuler24.03-py3.12

docker run \
    --name cann_container \
    --device /dev/davinci0 \
    --device /dev/davinci_manager \
    --device /dev/hisi_hdc \
    --device /dev/ummu \
    --device /dev/uburma \
    -v /usr/local/dcmi:/usr/local/dcmi \
    -v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi \
    -v /usr/local/Ascend/driver/lib64/:/usr/local/Ascend/driver/lib64/ \
    -v /usr/local/Ascend/driver/version.info:/usr/local/Ascend/driver/version.info \
    -v /etc/ascend_install.info:/etc/ascend_install.info \
    -it $CANN_REPO:$CANN_TAG bash

⁠How to Build Locally

git clone https://github.com/Ascend/cann-container-image.git
cd cann-container-image
export CANN_REPO=my-cann
export CANN_TAG=9.2.0-beta.2-a3-ubuntu22.04-py3.12
# need to install buildx
docker buildx build -t $CANN_REPO:$CANN_TAG -f cann/$CANN_TAG/Dockerfile .

⁠References

If you want to learn how to download CANN software packages, please visit the CANN download site⁠. If you want to learn more about CANN, including release notes, programming guides, APIs, and development tools, please visit the CANN documentation site⁠.

⁠License

View the license information⁠ for CANN software included in these images.

As with all container images, pre-installed packages (Python, system libraries, etc.) may be subject to their own licenses.