AI开发平台MODELARTS-训练基础镜像列表:训练基础镜像列表

时间:2024-04-30 17:31:41

训练基础镜像列表

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

表1 ModelArts训练基础镜像列表

引擎类型

版本名称

PyTorch

pytorch_1.8.0-cuda_10.2-py_3.7-ubuntu_18.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

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

support.huaweicloud.com/docker-modelarts/docker-modelarts_0022.html