Skip to content

Commit

Permalink
[cherry-pick][Fix][Sample-Yaml] Increase ray head CPU resource for py…
Browse files Browse the repository at this point in the history
…torch minst (#2330) (#2340)

Co-authored-by: Chi-Sheng Liu <[email protected]>
  • Loading branch information
kevin85421 and MortalHappiness authored Aug 27, 2024
1 parent 4dcdc8e commit dd3030a
Showing 1 changed file with 6 additions and 6 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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.
Expand All @@ -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
Expand All @@ -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"

0 comments on commit dd3030a

Please sign in to comment.