AI开发平台MODELARTS-准备模型训练镜像:训练作业的预置框架介绍

时间:2024-11-21 19:46:00

训练作业的预置框架介绍

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

support.huaweicloud.com/usermanual-standard-modelarts/develop-modelarts-1417.html