使用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容器镜像。

results matching ""

    No results matching ""