Class RaySpec (1.48.0)

RaySpec(mapping=None, *, ignore_unknown_fields=False, **kwargs)

Configuration information for the Ray cluster. For experimental launch, Ray cluster creation and Persistent cluster creation are 1:1 mapping: We will provision all the nodes within the Persistent cluster as Ray nodes.

Attributes

NameDescription
image_uri str
Optional. Default image for user to choose a preferred ML framework (for example, TensorFlow or Pytorch) by choosing from `Vertex prebuilt images
resource_pool_images MutableMapping[str, str]
Optional. Required if image_uri isn't set. A map of resource_pool_id to prebuild Ray image if user need to use different images for different head/worker pools. This map needs to cover all the resource pool ids. Example: { "ray_head_node_pool": "head image" "ray_worker_node_pool1": "worker image" "ray_worker_node_pool2": "another worker image" }
head_node_resource_pool_id str
Optional. This will be used to indicate which resource pool will serve as the Ray head node(the first node within that pool). Will use the machine from the first workerpool as the head node by default if this field isn't set.
ray_metric_spec google.cloud.aiplatform_v1beta1.types.RayMetricSpec
Optional. Ray metrics configurations.

Classes

ResourcePoolImagesEntry

ResourcePoolImagesEntry(mapping=None, *, ignore_unknown_fields=False, **kwargs)

The abstract base class for a message.

Parameters
NameDescription
kwargs dict

Keys and values corresponding to the fields of the message.

mapping Union[dict, .Message]

A dictionary or message to be used to determine the values for this message.

ignore_unknown_fields Optional(bool)

If True, do not raise errors for unknown fields. Only applied if mapping is a mapping type or there are keyword parameters.

Methods

RaySpec

RaySpec(mapping=None, *, ignore_unknown_fields=False, **kwargs)

Configuration information for the Ray cluster. For experimental launch, Ray cluster creation and Persistent cluster creation are 1:1 mapping: We will provision all the nodes within the Persistent cluster as Ray nodes.