# This config matches the default config for Anyscale workspaces with autoscaling,
# except instead of using CPU instances, it uses GPU instances.
cloud: {{env["ANYSCALE_CLOUD_NAME"]}}

advanced_instance_config:
    IamInstanceProfile: {"Name": "ray-autoscaler-v1"}

head_node:
    instance_type: m5.2xlarge

worker_nodes:
    # Anyscale workspaces use m5.2xlarge worker nodes by default. For consistency, we
    # use GPU nodes with the same number of vCPUs and memory.
    - instance_type: g4dn.2xlarge
      min_nodes: 0
      max_nodes: 10
      market_type: ON_DEMAND