使用GPU训练

简介

Xiaomi Cloud-ML支持使用GPU进行训练，只需在提交任务时指定GPU的数量即可，用户申请的GPU资源会通用容器进行映射和隔离，并且保证访问接口和宿主机GPU设备相同。

注意，用户需要有足够的GPU资源Quota才可以提交任务使用GPU。

使用GPU训练

用户编写TensorFlow模型代码后，提交时加入GPU参数即可使用GPU资源。

cloudml jobs submit -n linear -m trainer.task -u fds://cloud-ml/linear/trainer-1.0.tar.gz -c 1 -M 1G -g 1

参数介绍

• -g 参数表示使用的GPU数量，平台会选择资源充足的机器进行调度，除此之外启动的GPU容器镜像也不同于CPU容器镜像。