Class TpuClient (1.4.1)

Manages TPU nodes and other resources TPU API v2alpha1

Returns the transport used by the client instance.

Instantiates the tpu client.

Releases underlying transport's resources.

Returns a fully-qualified accelerator_type string.

Returns a fully-qualified billing_account string.

Returns a fully-qualified folder string.

Returns a fully-qualified location string.

Returns a fully-qualified organization string.

Returns a fully-qualified project string.

Creates a node.

from google.cloud import tpu_v2alpha1

def sample_create_node():
    # Create a client
    client = tpu_v2alpha1.TpuClient()

    # Initialize request argument(s)
    node = tpu_v2alpha1.Node()
    node.accelerator_type = "accelerator_type_value"
    node.runtime_version = "runtime_version_value"

    request = tpu_v2alpha1.CreateNodeRequest(
        parent="parent_value",
        node=node,
    )

    # Make the request
    operation = client.create_node(request=request)

    print("Waiting for operation to complete...")

    response = operation.result()

    # Handle the response
    print(response)

Deletes a node.

from google.cloud import tpu_v2alpha1

def sample_delete_node():
    # Create a client
    client = tpu_v2alpha1.TpuClient()

    # Initialize request argument(s)
    request = tpu_v2alpha1.DeleteNodeRequest(
        name="name_value",
    )

    # Make the request
    operation = client.delete_node(request=request)

    print("Waiting for operation to complete...")

    response = operation.result()

    # Handle the response
    print(response)

Creates an instance of this client using the provided credentials file.

Creates an instance of this client using the provided credentials info.

Creates an instance of this client using the provided credentials file.

Generates the Cloud TPU service identity for the project.

from google.cloud import tpu_v2alpha1

def sample_generate_service_identity():
    # Create a client
    client = tpu_v2alpha1.TpuClient()

    # Initialize request argument(s)
    request = tpu_v2alpha1.GenerateServiceIdentityRequest(
        parent="parent_value",
    )

    # Make the request
    response = client.generate_service_identity(request=request)

    # Handle the response
    print(response)

Gets AcceleratorType.

from google.cloud import tpu_v2alpha1

def sample_get_accelerator_type():
    # Create a client
    client = tpu_v2alpha1.TpuClient()

    # Initialize request argument(s)
    request = tpu_v2alpha1.GetAcceleratorTypeRequest(
        name="name_value",
    )

    # Make the request
    response = client.get_accelerator_type(request=request)

    # Handle the response
    print(response)

Retrieves the guest attributes for the node.

from google.cloud import tpu_v2alpha1

def sample_get_guest_attributes():
    # Create a client
    client = tpu_v2alpha1.TpuClient()

    # Initialize request argument(s)
    request = tpu_v2alpha1.GetGuestAttributesRequest(
        name="name_value",
    )

    # Make the request
    response = client.get_guest_attributes(request=request)

    # Handle the response
    print(response)

Return the API endpoint and client cert source for mutual TLS.

The client cert source is determined in the following order: (1) if GOOGLE_API_USE_CLIENT_CERTIFICATE environment variable is not "true", the client cert source is None. (2) if client_options.client_cert_source is provided, use the provided one; if the default client cert source exists, use the default one; otherwise the client cert source is None.

The API endpoint is determined in the following order: (1) if client_options.api_endpoint if provided, use the provided one. (2) if GOOGLE_API_USE_CLIENT_CERTIFICATE environment variable is "always", use the default mTLS endpoint; if the environment variabel is "never", use the default API endpoint; otherwise if client cert source exists, use the default mTLS endpoint, otherwise use the default API endpoint.

More details can be found at https://google.aip.dev/auth/4114.

Gets the details of a node.

from google.cloud import tpu_v2alpha1

def sample_get_node():
    # Create a client
    client = tpu_v2alpha1.TpuClient()

    # Initialize request argument(s)
    request = tpu_v2alpha1.GetNodeRequest(
        name="name_value",
    )

    # Make the request
    response = client.get_node(request=request)

    # Handle the response
    print(response)

Gets a runtime version.

from google.cloud import tpu_v2alpha1

def sample_get_runtime_version():
    # Create a client
    client = tpu_v2alpha1.TpuClient()

    # Initialize request argument(s)
    request = tpu_v2alpha1.GetRuntimeVersionRequest(
        name="name_value",
    )

    # Make the request
    response = client.get_runtime_version(request=request)

    # Handle the response
    print(response)

Lists accelerator types supported by this API.

from google.cloud import tpu_v2alpha1

def sample_list_accelerator_types():
    # Create a client
    client = tpu_v2alpha1.TpuClient()

    # Initialize request argument(s)
    request = tpu_v2alpha1.ListAcceleratorTypesRequest(
        parent="parent_value",
    )

    # Make the request
    page_result = client.list_accelerator_types(request=request)

    # Handle the response
    for response in page_result:
        print(response)

Lists nodes.

from google.cloud import tpu_v2alpha1

def sample_list_nodes():
    # Create a client
    client = tpu_v2alpha1.TpuClient()

    # Initialize request argument(s)
    request = tpu_v2alpha1.ListNodesRequest(
        parent="parent_value",
    )

    # Make the request
    page_result = client.list_nodes(request=request)

    # Handle the response
    for response in page_result:
        print(response)

Lists runtime versions supported by this API.

from google.cloud import tpu_v2alpha1

def sample_list_runtime_versions():
    # Create a client
    client = tpu_v2alpha1.TpuClient()

    # Initialize request argument(s)
    request = tpu_v2alpha1.ListRuntimeVersionsRequest(
        parent="parent_value",
    )

    # Make the request
    page_result = client.list_runtime_versions(request=request)

    # Handle the response
    for response in page_result:
        print(response)

Returns a fully-qualified node string.

Parses a accelerator_type path into its component segments.

Parse a billing_account path into its component segments.

Parse a folder path into its component segments.

Parse a location path into its component segments.

Parse a organization path into its component segments.

Parse a project path into its component segments.

Parses a node path into its component segments.

Parses a runtime_version path into its component segments.

Returns a fully-qualified runtime_version string.

Starts a node.

from google.cloud import tpu_v2alpha1

def sample_start_node():
    # Create a client
    client = tpu_v2alpha1.TpuClient()

    # Initialize request argument(s)
    request = tpu_v2alpha1.StartNodeRequest(
    )

    # Make the request
    operation = client.start_node(request=request)

    print("Waiting for operation to complete...")

    response = operation.result()

    # Handle the response
    print(response)

Stops a node. This operation is only available with single TPU nodes.

from google.cloud import tpu_v2alpha1

def sample_stop_node():
    # Create a client
    client = tpu_v2alpha1.TpuClient()

    # Initialize request argument(s)
    request = tpu_v2alpha1.StopNodeRequest(
    )

    # Make the request
    operation = client.stop_node(request=request)

    print("Waiting for operation to complete...")

    response = operation.result()

    # Handle the response
    print(response)

Updates the configurations of a node.

from google.cloud import tpu_v2alpha1

def sample_update_node():
    # Create a client
    client = tpu_v2alpha1.TpuClient()

    # Initialize request argument(s)
    node = tpu_v2alpha1.Node()
    node.accelerator_type = "accelerator_type_value"
    node.runtime_version = "runtime_version_value"

    request = tpu_v2alpha1.UpdateNodeRequest(
        node=node,
    )

    # Make the request
    operation = client.update_node(request=request)

    print("Waiting for operation to complete...")

    response = operation.result()

    # Handle the response
    print(response)