2gpus_64cpus_gce.yaml 987 B

123456789101112131415161718192021222324252627282930313233343536
  1. cloud_id: {{env["ANYSCALE_CLOUD_ID"]}}
  2. region: us-west1
  3. allowed_azs:
  4. - us-west1-b
  5. max_workers: 1
  6. # This head node was intentionally chosen to be larger than 2GPUs (has 4GPUs).
  7. # This makes sure we can safely migrate the 2GPU testcases to GCE without having
  8. # 2GPU instances available on GCE today.
  9. # TODO (Artur): Move to 2GPU head node once it's available or possibly make tests using this use 4 GPUs.
  10. head_node_type:
  11. name: head_node
  12. instance_type: n1-highmem-32-nvidia-tesla-v100-4 # g3.8xlarge
  13. worker_node_types:
  14. - name: worker_node
  15. instance_type: n2-standard-32 # m5.8xlarge
  16. min_workers: 1
  17. max_workers: 1
  18. use_spot: false
  19. gcp_advanced_configurations_json:
  20. instance_properties:
  21. disks:
  22. - boot: true
  23. auto_delete: true
  24. initialize_params:
  25. disk_size_gb: 500
  26. #aws:
  27. # BlockDeviceMappings:
  28. # - DeviceName: /dev/sda1
  29. # Ebs:
  30. # DeleteOnTermination: true
  31. # VolumeSize: 500