Help Center/ ModelArts/ ModelArts User Guide (Standard)/ Creating a Custom Image for ModelArts Standard/ Preset Images Supported by ModelArts/ Preset Dedicated Images for Training
Updated on 2025-11-18 GMT+08:00
View PDF
Share

Preset Dedicated Images for Training

ModelArts provides deep learning-powered base images such as TensorFlow, PyTorch, and MindSpore images. In these images, the software mandatory for running training jobs has been installed. If the software in the base images cannot meet your service requirements, create new images based on the base images and use the new images to create training jobs.

Available Training Base Images

The following table lists the preset training base images of ModelArts.

Table 1 ModelArts training base images

Engine

Version

PyTorch

pytorch_1.8.0-cuda_10.2-py_3.7-ubuntu_18.04-x86_64

pytorch_2.1.0-cuda_12.1-py_3.10.6-ubuntu_22.04-x86_64

TensorFlow

tensorflow_2.1.0-cuda_10.1-py_3.7-ubuntu_18.04-x86_64

Horovod

horovod_0.20.0-tensorflow_2.1.0-cuda_10.1-py_3.7-ubuntu_18.04-x86_64

horovod_0.22.1-pytorch_1.8.0-cuda_10.2-py_3.7-ubuntu_18.04-x86_64

MPI

mindspore_1.3.0-cuda_10.1-py_3.7-ubuntu_1804-x86_64

Ascend-Powered-Engine

tensorflow_1.15-cann_5.1.0-py_3.7-euler_2.8.3-aarch64

mindspore_1.7.0-cann_5.1.0-py_3.7-euler_2.8.3-aarch64

pytorch_2.1.0-cann_7.0.1.1-py_3.9-euler_2.10.7-aarch64-snt3p

mindspore_2.2.12-cann_7.0.1.1-py_3.9-euler_2.10.7-aarch64-snt3p

Supported AI engines vary depending on regions.

PyTorch Base Images

This section describes preset PyTorch images.

Engine Version 1: pytorch_2.1.0-cuda_12.1-py_3.10.6-ubuntu_22.04-x86_64

Engine Version 2: pytorch_1.8.0-cuda_10.2-py_3.7-ubuntu_18.04-x86_64

Engine Version 1: pytorch_2.1.0-cuda_12.1-py_3.10.6-ubuntu_22.04-x86_64

  • Image path: swr.{<region-ID>}.myhuaweicloud.com/atelier/pytorch_2_1:pytorch_2.1.0-cuda_12.1-py_3.10.6-ubuntu_22.04-x86_64-20250305173557-cb53968
  • Image creation time: 20250305173557 (yyyy-mm-dd-hh-mm-ss)
  • Image system version: Ubuntu 22.04 LTS
  • CUDA: 12.1
  • Path and version of the Python interpreter: /home/ma-user/anaconda3/envs/PyTorch-2.1/bin/python, Python 3.10.6
  • Third-party package installation path: /home/ma-user/anaconda3/envs/PyTorch-2.1/lib/python3.10.6/site-packages

Engine Version 2: pytorch_1.8.0-cuda_10.2-py_3.7-ubuntu_18.04-x86_64

  • Image address: swr.{Region}.myhuaweicloud.com/aip/pytorch_1_8:train-pytorch_1.8.0-cuda_10.2-py_3.7-ubuntu_18.04-x86_64-roma-20220309171256-40adcc1
  • Image creation time: 20220309171256 (yyyy-mm-dd-hh-mm-ss)
  • Image system version: Ubuntu 18.04.4 LTS
  • CUDA: 10.2.89
  • cuDNN: 7.6.5.32
  • Path and version of the Python interpreter: /home/ma-user/anaconda3/envs/PyTorch-1.8/bin/python, Python 3.7.10
  • Third-party package installation path: /home/ma-user/anaconda3/envs/PyTorch-1.8/lib/python3.7/site-packages

TensorFlow Base Images

This section describes preset TensorFlow images.

Engine Version: tensorflow_2.1.0-cuda_10.1-py_3.7-ubuntu_18.04-x86_64

Engine Version: tensorflow_2.1.0-cuda_10.1-py_3.7-ubuntu_18.04-x86_64

  • Image address: swr.{Region}.myhuaweicloud.com/aip/tensorflow_2_1:train-tensorflow_2.1.0-cuda_10.1-py_3.7-ubuntu_18.04-x86_64-20210912152543-1e0838d
  • Image creation time: 20210912152543(yyyy-mm-dd-hh-mm-ss)
  • Image system version: Ubuntu 18.04.4 LTS
  • CUDA: 10.1.243
  • cuDNN: 7.6.5.32
  • Path and version of the Python interpreter: /home/ma-user/anaconda3/envs/TensorFlow-2.1/bin/python, Python 3.7.10
  • Third-party package installation path: /home/ma-user/anaconda3/envs/TensorFlow-2.1/lib/python3.7/site-packages

Horovod Base Images

This section describes preset Horovod images.

Engine Version 1: horovod_0.20.0-tensorflow_2.1.0-cuda_10.1-py_3.7-ubuntu_18.04-x86_64

  • Image address: swr.{Region}.myhuaweicloud.com/aip/horovod_tensorflow:train-horovod_0.20.0-tensorflow_2.1.0-cuda_10.1-py_3.7-ubuntu_18.04-x86_64-20210912152543-1e0838d
  • Image creation time: 20210912152543 (yyyy-mm-dd-hh-mm-ss)
  • Image system version: Ubuntu 18.04.4 LTS
  • CUDA: 10.1.243
  • cuDNN: 7.6.5.32
  • Path and version of the Python interpreter: /home/ma-user/anaconda3/envs/horovod_0.20.0-tensorflow_2.1.0/bin/python, Python 3.7.10
  • Third-party package installation path: /home/ma-user/anaconda3/envs/horovod_0.20.0-tensorflow_2.1.0/lib/python3.7/site-packages

Engine Version 2: horovod_0.22.1-pytorch_1.8.0-cuda_10.2-py_3.7-ubuntu_18.04-x86_64

  • Image address: swr.{Region}.myhuaweicloud.com/aip/horovod_pytorch:train-horovod_0.22.1-pytorch_1.8.0-cuda_10.2-py_3.7-ubuntu_18.04-x86_64-20210912152543-1e0838d
  • Image creation time: 20210912152543 (yyyy-mm-dd-hh-mm-ss)
  • Image system version: Ubuntu 18.04.4 LTS
  • CUDA: 11.1.1
  • cuDNN: 8.0.5.39
  • Path and version of the Python interpreter: /home/ma-user/anaconda3/envs/horovod-0.22.1-pytorch-1.8.0/bin/python, Python 3.7.10
  • Third-party package installation path: /home/ma-user/anaconda3/envs/horovod-0.22.1-pytorch-1.8.0/lib/python3.7/site-packages

MPI Base Images

This section describes preset mindspore_1.3.0 images.

Engine Version: mindspore_1.3.0-cuda_10.1-py_3.7-ubuntu_1804-x86_64

Engine Version: mindspore_1.3.0-cuda_10.1-py_3.7-ubuntu_1804-x86_64

  • Image address: swr.{Region}.myhuaweicloud.com/aip/mindspore_1_3_0:train-mindspore_1.3.0-cuda_10.1-py_3.7-ubuntu_18.04-x86_64-roma-20211104202338-f258e59
  • Image creation time: 20211104202338 (yyyy-mm-dd-hh-mm-ss)
  • Image system version: Ubuntu 18.04.4 LTS
  • CUDA: 10.1.243
  • cuDNN: 7.6.5.32
  • Path and version of the Python interpreter: /home/ma-user/anaconda3/envs/MindSpore-1.3.0-gpu/bin/python, Python 3.7.10
  • Third-party package installation path: /home/ma-user/anaconda3/envs/MindSpore-1.3.0-gpu/lib/python3.7/site-packages

Training Base Image (Ascend-Powered-Engine)

This section describes preset Ascend-Powered-Engine images.

Engine Version 1: pytorch_2.1.0-cann_7.0.1.1-py_3.9-euler_2.10.7-aarch64-snt3p

  • Image path: swr.<region>.myhuaweicloud.com/atelier/pytorch_2_1_ascend:pytorch_2.1.0-cann_7.0.1.1-py_3.9-euler_2.10.7-aarch64-snt3p-20240520103850-0d60076
  • Image creation time: 20240520103850 (yyyy-mm-dd-hh-mm-ss)
  • Image system version: EulerOS 2.10.7
  • MindSpore Lite: 2.2.12
  • Ascend CANN Toolkit: 7.0.1.1

Engine Version 2: mindspore_2.2.12-cann_7.0.1.1-py_3.9-euler_2.10.7-aarch64-snt3p

  • Image path: swr.<<region>>.myhuaweicloud.com/atelier/mindspore_2_2_ascend:mindspore_2.2.12-cann_7.0.1.1-py_3.9-euler_2.10.7-aarch64-snt3p-20240520103850-0d60076
  • Image creation time: 20240520103850 (yyyy-mm-dd-hh-mm-ss)
  • Image system version: EulerOS 2.10.7
  • MindSpore Lite: 2.2.12
  • Ascend CANN Toolkit: 7.0.1.1

Engine Version 3: tensorflow_1.15-cann_5.1.0-py_3.7-euler_2.8.3-aarch64

  • Image path: swr.<<region>>.myhuaweicloud.com/atelier/tensorflow_1_15_ascend:tensorflow_1.15-cann_5.1.0-py_3.7-euler_2.8.3-aarch64-d910-20220715093657-9446c6a
  • Image creation time: 20220715093657 (yyyy-mm-dd-hh-mm-ss)
  • Image system version: EulerOS 2.8.3
  • TensorFlow: 1.15
  • Ascend CANN Toolkit: 5.1.0

Engine Version 4: mindspore_1.7.0-cann_5.1.0-py_3.7-euler_2.8.3-aarch64

  • Image path: swr.<<region>>.myhuaweicloud.com/atelier/mindspore_1_7_0:mindspore_1.7.0-cann_5.1.0-py_3.7-euler_2.8.3-aarch64-d910-20220715093657-9446c6a
  • Image creation time: 20220715093657 (yyyy-mm-dd-hh-mm-ss)
  • Image system version: EulerOS 2.8.3
  • MindSpore Lite: 1.7.0
  • Ascend CANN Toolkit: 5.1.0