Skip to content

Commit

Permalink
[Fix][Sample-Yaml] Increase ray head CPU resource for pytorch minst (r…
Browse files Browse the repository at this point in the history
  • Loading branch information
MortalHappiness authored Aug 26, 2024
1 parent e7f0c2c commit a69252e
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 a69252e

Please sign in to comment.