华为云AI开发平台ModelArts训练基础镜像列表_云淘科技

11 月 08, 2023

104 0

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

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引擎有差异，请以实际环境为准。

父主题： 训练基础镜像详情介绍

内容没看懂？不太想学习？想快速解决？有偿解决：联系专家

分类