更新时间:2024-04-30 GMT+08:00
训练基础镜像列表
ModelArts平台提供了Tensorflow,PyTorch,MindSpore等常用深度学习任务的基础镜像,镜像里已经安装好运行任务所需软件。当基础镜像里的软件无法满足您的程序运行需求时,您可以基于这些基础镜像制作一个新的镜像并进行训练。
训练基础镜像列表
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 |
![](https://support.huaweicloud.com/intl/zh-cn/docker-modelarts/public_sys-resources/note_3.0-zh-cn.png)
不同区域支持的AI引擎有差异,请以实际环境为准。
父主题: 训练基础镜像详情介绍