From a69252e6cc90995a3b3085f14af5bcc6869a2d93 Mon Sep 17 00:00:00 2001
From: Chi-Sheng Liu <chishengliu@chishengliu.com>
Date: Mon, 26 Aug 2024 22:43:58 +0800
Subject: [PATCH] [Fix][Sample-Yaml] Increase ray head CPU resource for pytorch
 minst (#2330)

---
 .../samples/pytorch-mnist/ray-job.pytorch-mnist.yaml | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/ray-operator/config/samples/pytorch-mnist/ray-job.pytorch-mnist.yaml b/ray-operator/config/samples/pytorch-mnist/ray-job.pytorch-mnist.yaml
index 62804e9739..818cdfc19b 100644
--- a/ray-operator/config/samples/pytorch-mnist/ray-job.pytorch-mnist.yaml
+++ b/ray-operator/config/samples/pytorch-mnist/ray-job.pytorch-mnist.yaml
@@ -11,7 +11,7 @@ spec:
       - torchvision
     working_dir: "https://github.com/ray-project/kuberay/archive/master.zip"
     env_vars:
-      NUM_WORKERS: "4"
+      NUM_WORKERS: "2"
       CPUS_PER_WORKER: "2"
 
   # rayClusterSpec specifies the RayCluster instance to be created by the RayJob controller.
@@ -34,13 +34,13 @@ spec:
                   name: client
               resources:
                 limits:
-                  cpu: "2"
+                  cpu: "1"
                   memory: "4Gi"
                 requests:
-                  cpu: "2"
+                  cpu: "1"
                   memory: "4Gi"
     workerGroupSpecs:
-      - replicas: 4
+      - replicas: 2
         minReplicas: 1
         maxReplicas: 5
         groupName: small-group
@@ -53,8 +53,8 @@ spec:
                 image: rayproject/ray:2.34.0
                 resources:
                   limits:
-                    cpu: "2"
+                    cpu: "3"
                     memory: "4Gi"
                   requests:
-                    cpu: "2"
+                    cpu: "3"
                     memory: "4Gi"