创建算法
对于PyTorch模型,需要上传自定义镜像创建算法,具体操作步骤如下:
- 参考镜像获取章节,从AscendHub镜像中心获取云上训练所需镜像ascend-pytorch-arm-modelarts(最新版本)。
- 参考镜像使用章节,以ma-user用户登录基础镜像拉起的容器,安装CRNN所需的三方库依赖:
pip install -r 1.5_requirements.txt # 基础镜像torch版本为1.5,安装对应版本依赖
- 退出容器,参考如下命令将容器提交为新的镜像:
docker commit crnn crnn_modelarts:0.0.1 # 将名为crnn的容器提交成新镜像,其中镜像名称为crnn_modelarts,tag为0.0.1
- 登录SWR控制台,将准备好的镜像上传。上传方法参见《容器镜像服务 SWR》“快速入门 > 使用容器引擎客户端上传镜像”章节。
SWR的区域需要与ModelArts所在的区域一致。例如:当前ModelArts在华北-北京四区域,SWR所在区域,请选择华北-北京四。