获取训练镜像
可选择以下方式中的一种来获取训练镜像:
- (推荐)从昇腾镜像仓库根据系统架构(Arm/x86_64)、模型框架(TensorFlow、PyTorch、MindSpore)下载配套驱动版本的训练基础镜像。基于训练基础镜像进行修改,将容器中默认用户修改为root(21.0.4版本之后训练基础镜像默认用户为非root)。基础镜像中不包含训练脚本、代码等文件,训练时通常使用挂载的方式将训练脚本、代码等文件映射到容器内。
- 从头开始定制用户自己的训练镜像,制作过程请参考制作镜像中制作容器相关章节。
可将下载/制作的训练基础镜像重命名,如:training:v7.0.RC1。
加固镜像
下载或者制作的训练基础镜像可以进行安全加固,提升镜像安全性,可参见容器安全加固章节进行操作。