Ascend Docker Images

中文 • Docker Hub • GitHub Container • Red Hat Quay

This repository hosts the following Ascend Docker images, providing base images based on both Ubuntu and OpenEuler. For detailed information, please refer to the README.md file in each image directory:

• CANN：provides cann-toolkit runtime environment
• PyTorch: provides torch_npu runtime environment
• MindSpore: provides MindSpore runtime environment
• Python: provides a base Python environment

Usage

Please make sure you have an NPU device and the driver has been installed, then refer to the docker run commands in the image directory's README.md.

Build

We recommend using Docker Buildx Bake for building the images. For build details, see the docker-bake.hcl, and for the build parameters, check arg.json.

Note

If you don't have Bake, you can find the traditional docker build commands in the image directory's README.md.

To build using Bake, Docker Engine 20.10+ is required. Run the following command in the root directory of this repository:

docker buildx bake -f arg.json -f docker-bake.hcl

To list all targets:

docker buildx bake -f arg.json -f docker-bake.hcl --print

To build single-arch images only:

docker buildx bake -f arg.json -f docker-bake.hcl \
    --set '*.platform=linux/arm64'

Note

• To customize the build configuration, use the --set. More details here: https://docs.docker.com/reference/cli/docker/buildx/bake/#set
• To modify build parameters or tag names, edit the arg.json file

Support

The team tracks bugs and enhancement requests using GitHub issues. Before submitting a suggestion or bug report, search the existing GitHub issues to see if your issue has already been reported.

License

This repository is licensed under the Apache License 2.0.