使用GPU开发环境

简介

Xiaomi Cloud-ML也支持基于GPU的开发环境，用法与训练任务相同，可参考 使用GPU训练 。开发环境的GPU数量大于0时，开启GPU加速。系统为开启GPU的开发环境分配GPU，并且镜像内安装相应软件包。

使用命令行启用GPU

用户在使用命令行创建开发环境时，可以使用参数指定需要的GPU个数。

  1. cloudml dev create -n mydev -p mypassword -c 1 -M 1G -g 1 -d cnbj6-repo.cloud.mi.com/cloud-ml/tensorflow-gpu:1.7.0-xm1.0.0
  1. cloudml dev create-n mydev -p mypassword -c 1 -M 1G -g -gt p40 -d cnbj6-repo.cloud.mi.com/cloud-ml/tensorflow-gpu:1.7.0-xm1.0.0
  • -g 参数表示使用的GPU数量，平台会选择资源充足的机器进行调度，除此之外启动的GPU容器镜像也不同于CPU容器镜像。
  • -gt 可选参数，可以指定GPU的类型，目前支持p40, m40, k40。用户指定了该参数时，系统会按照指定类型的GPU创建实例，否则随机分配任意可使用的GPU。

    使用WEB控制台启用GPU

用户在使用WEB控制台创建开发环境时，可以在页面中选择GPU数量，指定需要的GPU个数。dev gpu

原文: http://docs.api.xiaomi.com/cloud-ml/devenv/0401_use_gpu.html