您可以新建一个 PaddlePaddle 类型的任务。
前提条件
- 您已成功安装 CCE AI Job Scheduler 和 CCE Deep Learning Frameworks Operator 组件,否则云原生 AI 功能将无法使用。
- 若您是子用户,队列关联的用户中有您才能使用该队列新建任务。
- 安装组件 CCE Deep Learning Frameworks Operator 时,系统安装了 PaddlePaddle 深度学习框架。
限制说明
- 目前 PaddlePaddle 类型的任务不支持 GPU 显存共享。
操作步骤
- 登录百度智能云官网,并进入管理控制台。
- 选择“产品服务 > 云原生 > 容器引擎 CCE”,单击进入容器引擎管理控制台。
- 单击左侧导航栏中的 集群管理 > 集群列表 。
- 在集群列表页面中,单击目标集群名称进入集群管理页面。
- 在集群管理页面单击 云原生AI > 任务管理 。
- 在任务管理页面单击 新建任务 。
- 在新建任务页面中,完成任务基本信息配置:
- 任务名称:自定义任务名称,支持大小写字母、数字、以及-_ /.特殊字符,必须以中文或字母开头,长度 1-65。
- 队列:选择新建任务关联的队列。
- 框架:选择任务对应的深度学习框架“PaddlePaddle”。
- 参考 yaml 模板完成配置:
apiVersion: batch.paddlepaddle.org/v1
kind: PaddleJob
metadata:
name: resnet
spec:
cleanPodPolicy: Never
worker:
replicas: 2
template:
spec:
schedulerName: volcano
containers:
- name: resnet
image: registry.baidubce.com/paddle-operator/demo-resnet:v1
command:
- python
args:
- "-m"
- "paddle.distributed.launch"
- "train_fleet.py"
volumeMounts:
- mountPath: /dev/shm
name: dshm
resources:
requests:
cpu: 1
memory: 2Gi
limits:
baidu.com/v100_16g_cgpu: "1"
volumes:
- name: dshm
emptyDir:
medium: Memory
- 点击“确定”按钮,完成任务的新建。