定制训练任务Docker镜像

简介

Xiaomi Cloud-ML支持Bring Your Own Image功能,允许用户传入自定义的Docker镜像来启动训练任务,可以解决训练任务依赖非Python库的问题。

注意,用户提供的镜像地址应该是公开可访问的,如果在DockerHub中下载镜像和启动任务时间较长。

使用示例

用户提交任务时指定Docker镜像地址即可。

cloudml jobs submit -n linear -m trainer.task -u fds://cloud-ml/linear/trainer-1.0.tar.gz -d cloudml/tensorflow:1.0.0

参数介绍

  • -d 表示用户指定的Docker镜像地址,注意不可与 -F-V 参数同时使用。

results matching ""

    No results matching ""