使用GPU开发环境

简介

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

使用命令行启用GPU

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

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
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

results matching ""

    No results matching ""