更新时间:2025-08-07 GMT+08:00

训练专属预置镜像列表

ModelArts平台提供了Tensorflow,PyTorch,MindSpore等常用深度学习任务的基础镜像,镜像里已经安装好运行任务所需软件。当基础镜像里的软件无法满足您的程序运行需求时,您可以基于这些基础镜像制作一个新的镜像并进行训练。

训练基础镜像列表

ModelArts中预置的训练基础镜像如下表所示。

表1 ModelArts训练基础镜像列表

引擎类型

版本名称

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

不同区域支持的AI引擎有差异,请以实际环境为准。

训练基础镜像详情(TensorFlow)

介绍预置的TensorFlow镜像详情。

训练基础镜像详情(MPI)

介绍预置的mindspore_1.3.0镜像详情。

引擎版本:mindspore_1.3.0-cuda_10.1-py_3.7-ubuntu_1804-x86_64

引擎版本一:pytorch_2.1.0-cann_7.0.1.1-py_3.9-euler_2.10.7-aarch64-snt3p

  • 镜像地址: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
  • 镜像构建时间:20240520103850(yyyy-mm-dd-hh-mm-ss)
  • 镜像系统版本:euler 2.10.7
  • mindspore-lite:2.2.12
  • Ascend CANN Toolkit:7.0.1.1

引擎版本二:mindspore_2.2.12-cann_7.0.1.1-py_3.9-euler_2.10.7-aarch64-snt3p

  • 镜像地址: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
  • 镜像构建时间:20240520103850(yyyy-mm-dd-hh-mm-ss)
  • 镜像系统版本:euler 2.10.7
  • mindspore-lite:2.2.12
  • Ascend CANN Toolkit:7.0.1.1

引擎版本三:tensorflow_1.15-cann_5.1.0-py_3.7-euler_2.8.3-aarch64

  • 镜像地址: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
  • 镜像构建时间:20220715093657(yyyy-mm-dd-hh-mm-ss)
  • 镜像系统版本:euler 2.8.3
  • tensorflow:1.15
  • Ascend CANN Toolkit:5.1.0

引擎版本四:mindspore_1.7.0-cann_5.1.0-py_3.7-euler_2.8.3-aarch64

  • 镜像地址: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
  • 镜像构建时间:20220715093657(yyyy-mm-dd-hh-mm-ss)
  • 镜像系统版本:euler 2.8.3
  • mindspore-lite:1.7.0
  • Ascend CANN Toolkit:5.1.0