AI开发平台MODELARTS-准备模型训练镜像:构建自定义训练镜

时间:2024-08-16 20:38:58

构建自定义训练镜

当基础镜像里的软件无法满足您的程序运行需求时,您还可以基于这些基础镜像制作一个新的镜像并进行训练。镜像制作流程如图1所示。

图1 训练作业的 自定义镜像 制作流程

场景一:预置镜像满足ModelArts训练平台约束,但不满足代码依赖的要求,需要额外安装软件包。

具体案例参考使用预置镜像制作自定义镜像用于训练模型。

场景二:已有本地镜像满足代码依赖的要求,但是不满足ModelArts训练平台约束,需要适配。

具体案例参考已有镜像迁移至ModelArts用于训练模型。

场景三: 当前无可使用的镜像,需要从0制作镜像(既需要安装代码依赖,又需要制作出的镜像满足MAModelArts平台约束)。具体案例参考:

  • 从0制作自定义镜像用于创建训练作业(PyTorch+CPU/GPU)
  • 从0制作自定义镜像用于创建训练作业(MPI+CPU/GPU)
  • 从0制作自定义镜像用于创建训练作业(Tensorflow+GPU)
  • 从0制作自定义镜像用于创建训练作业(MindSpore+Ascend)
support.huaweicloud.com/usermanual-standard-modelarts/develop-modelarts-1417.html