使用其他深度学习框架

简介

Xiaomi Cloud-ML支持几乎所有主流的深度学习框架,用户只需要编写对应的应用代码,提交任务时指定选择的框架和版本即可。

要查询当前服务支持的深度学习框架,可以使用下面的命令。

cloudml framework list

使用其他框架

例如我们编写了MXNet模型代码,参考前面的方式打包并上传到FDS中,然后使用下面的命令提交即可。

cloudml jobs submit -n linear -m trainer.task -u fds://cloud-ml/linear/trainer-1.0.tar.gz -F mxnet -V 0.9.0

或者通过JSON文件的方式定义,编写 mxnet_job.json 文件。

{
  "job_name": "mxnet",
  "module_name": "trainer.task",
  "trainer_uri": "fds://cloud-ml/trainer-1.0.tar.gz",
  "job_args": "",
  "cpu_limit": "0.5",
  "memory_limit": "100M",
  "framework": "mxnet",
  "framework_version": "0.9.0"
}

然后使用命令行提交,同样可以使用GPU、NodeSelector等参数。

cloudml jobs submit -f ./mxnet_job.json

使用其他版本

TensorFlow深度学习框架目前支持 0.12.01.0.0 版本,未来可能支持更多版本的API,提交任务时可以直接指定对应的框架和版本。

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

参数介绍

  • -F 表示使用的深度学习框架。
  • -V 表示使用深度学习框架的版本。

results matching ""

    No results matching ""