Skip to content

Commit dd3030a

Browse files
[cherry-pick][Fix][Sample-Yaml] Increase ray head CPU resource for pytorch minst (#2330) (#2340)
Co-authored-by: Chi-Sheng Liu <[email protected]>
1 parent 4dcdc8e commit dd3030a

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

ray-operator/config/samples/pytorch-mnist/ray-job.pytorch-mnist.yaml

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@ spec:
1111
- torchvision
1212
working_dir: "https://github.com/ray-project/kuberay/archive/master.zip"
1313
env_vars:
14-
NUM_WORKERS: "4"
14+
NUM_WORKERS: "2"
1515
CPUS_PER_WORKER: "2"
1616
1717
# rayClusterSpec specifies the RayCluster instance to be created by the RayJob controller.
@@ -34,13 +34,13 @@ spec:
3434
name: client
3535
resources:
3636
limits:
37-
cpu: "2"
37+
cpu: "1"
3838
memory: "4Gi"
3939
requests:
40-
cpu: "2"
40+
cpu: "1"
4141
memory: "4Gi"
4242
workerGroupSpecs:
43-
- replicas: 4
43+
- replicas: 2
4444
minReplicas: 1
4545
maxReplicas: 5
4646
groupName: small-group
@@ -53,8 +53,8 @@ spec:
5353
image: rayproject/ray:2.34.0
5454
resources:
5555
limits:
56-
cpu: "2"
56+
cpu: "3"
5757
memory: "4Gi"
5858
requests:
59-
cpu: "2"
59+
cpu: "3"
6060
memory: "4Gi"

0 commit comments

Comments
 (0)