TpuClient(*, credentials: Optional[google.auth.credentials.Credentials] = None, transport: Optional[Union[str, google.cloud.tpu_v1.services.tpu.transports.base.TpuTransport]] = None, client_options: Optional[google.api_core.client_options.ClientOptions] = None, client_info: google.api_core.gapic_v1.client_info.ClientInfo = <google.api_core.gapic_v1.client_info.ClientInfo object>)
Manages TPU nodes and other resources TPU API v1
Properties
transport
Returns the transport used by the client instance.
Returns | |
---|---|
Type | Description |
TpuTransport | The transport used by the client instance. |
Methods
TpuClient
TpuClient(*, credentials: Optional[google.auth.credentials.Credentials] = None, transport: Optional[Union[str, google.cloud.tpu_v1.services.tpu.transports.base.TpuTransport]] = None, client_options: Optional[google.api_core.client_options.ClientOptions] = None, client_info: google.api_core.gapic_v1.client_info.ClientInfo = <google.api_core.gapic_v1.client_info.ClientInfo object>)
Instantiates the tpu client.
Parameters | |
---|---|
Name | Description |
credentials |
Optional[google.auth.credentials.Credentials]
The authorization credentials to attach to requests. These credentials identify the application to the service; if none are specified, the client will attempt to ascertain the credentials from the environment. |
transport |
Union[str, TpuTransport]
The transport to use. If set to None, a transport is chosen automatically. |
client_options |
google.api_core.client_options.ClientOptions
Custom options for the client. It won't take effect if a |
client_info |
google.api_core.gapic_v1.client_info.ClientInfo
The client info used to send a user-agent string along with API requests. If |
Exceptions | |
---|---|
Type | Description |
google.auth.exceptions.MutualTLSChannelError | If mutual TLS transport creation failed for any reason. |
__exit__
__exit__(type, value, traceback)
Releases underlying transport's resources.
accelerator_type_path
accelerator_type_path(project: str, location: str, accelerator_type: str)
Returns a fully-qualified accelerator_type string.
common_billing_account_path
common_billing_account_path(billing_account: str)
Returns a fully-qualified billing_account string.
common_folder_path
common_folder_path(folder: str)
Returns a fully-qualified folder string.
common_location_path
common_location_path(project: str, location: str)
Returns a fully-qualified location string.
common_organization_path
common_organization_path(organization: str)
Returns a fully-qualified organization string.
common_project_path
common_project_path(project: str)
Returns a fully-qualified project string.
create_node
create_node(request: Optional[Union[google.cloud.tpu_v1.types.cloud_tpu.CreateNodeRequest, dict]] = None, *, parent: Optional[str] = None, node: Optional[google.cloud.tpu_v1.types.cloud_tpu.Node] = None, node_id: Optional[str] = None, retry: Union[google.api_core.retry.Retry, google.api_core.gapic_v1.method._MethodDefault] = <_MethodDefault._DEFAULT_VALUE: <object object>>, timeout: Optional[float] = None, metadata: Sequence[Tuple[str, str]] = ())
Creates a node.
# This snippet has been automatically generated and should be regarded as a
# code template only.
# It will require modifications to work:
# - It may require correct/in-range values for request initialization.
# - It may require specifying regional endpoints when creating the service
# client as shown in:
# https://googleapis.dev/python/google-api-core/latest/client_options.html
from google.cloud import tpu_v1
def sample_create_node():
# Create a client
client = tpu_v1.TpuClient()
# Initialize request argument(s)
node = tpu_v1.Node()
node.accelerator_type = "accelerator_type_value"
node.tensorflow_version = "tensorflow_version_value"
request = tpu_v1.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)
Parameters | |
---|---|
Name | Description |
request |
Union[google.cloud.tpu_v1.types.CreateNodeRequest, dict]
The request object. Request for CreateNode. |
parent |
str
Required. The parent resource name. This corresponds to the |
node |
google.cloud.tpu_v1.types.Node
Required. The node. This corresponds to the |
node_id |
str
The unqualified resource name. This corresponds to the |
retry |
google.api_core.retry.Retry
Designation of what errors, if any, should be retried. |
timeout |
float
The timeout for this request. |
metadata |
Sequence[Tuple[str, str]]
Strings which should be sent along with the request as metadata. |
Returns | |
---|---|
Type | Description |
google.api_core.operation.Operation | An object representing a long-running operation. The result type for the operation will be Node A TPU instance. |
delete_node
delete_node(request: Optional[Union[google.cloud.tpu_v1.types.cloud_tpu.DeleteNodeRequest, dict]] = None, *, name: Optional[str] = None, retry: Union[google.api_core.retry.Retry, google.api_core.gapic_v1.method._MethodDefault] = <_MethodDefault._DEFAULT_VALUE: <object object>>, timeout: Optional[float] = None, metadata: Sequence[Tuple[str, str]] = ())
Deletes a node.
# This snippet has been automatically generated and should be regarded as a
# code template only.
# It will require modifications to work:
# - It may require correct/in-range values for request initialization.
# - It may require specifying regional endpoints when creating the service
# client as shown in:
# https://googleapis.dev/python/google-api-core/latest/client_options.html
from google.cloud import tpu_v1
def sample_delete_node():
# Create a client
client = tpu_v1.TpuClient()
# Initialize request argument(s)
request = tpu_v1.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)
Parameters | |
---|---|
Name | Description |
request |
Union[google.cloud.tpu_v1.types.DeleteNodeRequest, dict]
The request object. Request for DeleteNode. |
name |
str
Required. The resource name. This corresponds to the |
retry |
google.api_core.retry.Retry
Designation of what errors, if any, should be retried. |
timeout |
float
The timeout for this request. |
metadata |
Sequence[Tuple[str, str]]
Strings which should be sent along with the request as metadata. |
Returns | |
---|---|
Type | Description |
google.api_core.operation.Operation | An object representing a long-running operation. The result type for the operation will be Node A TPU instance. |
from_service_account_file
from_service_account_file(filename: str, *args, **kwargs)
Creates an instance of this client using the provided credentials file.
Parameter | |
---|---|
Name | Description |
filename |
str
The path to the service account private key json file. |
Returns | |
---|---|
Type | Description |
TpuClient | The constructed client. |
from_service_account_info
from_service_account_info(info: dict, *args, **kwargs)
Creates an instance of this client using the provided credentials info.
Parameter | |
---|---|
Name | Description |
info |
dict
The service account private key info. |
Returns | |
---|---|
Type | Description |
TpuClient | The constructed client. |
from_service_account_json
from_service_account_json(filename: str, *args, **kwargs)
Creates an instance of this client using the provided credentials file.
Parameter | |
---|---|
Name | Description |
filename |
str
The path to the service account private key json file. |
Returns | |
---|---|
Type | Description |
TpuClient | The constructed client. |
get_accelerator_type
get_accelerator_type(request: Optional[Union[google.cloud.tpu_v1.types.cloud_tpu.GetAcceleratorTypeRequest, dict]] = None, *, name: Optional[str] = None, retry: Union[google.api_core.retry.Retry, google.api_core.gapic_v1.method._MethodDefault] = <_MethodDefault._DEFAULT_VALUE: <object object>>, timeout: Optional[float] = None, metadata: Sequence[Tuple[str, str]] = ())
Gets AcceleratorType.
# This snippet has been automatically generated and should be regarded as a
# code template only.
# It will require modifications to work:
# - It may require correct/in-range values for request initialization.
# - It may require specifying regional endpoints when creating the service
# client as shown in:
# https://googleapis.dev/python/google-api-core/latest/client_options.html
from google.cloud import tpu_v1
def sample_get_accelerator_type():
# Create a client
client = tpu_v1.TpuClient()
# Initialize request argument(s)
request = tpu_v1.GetAcceleratorTypeRequest(
name="name_value",
)
# Make the request
response = client.get_accelerator_type(request=request)
# Handle the response
print(response)
Parameters | |
---|---|
Name | Description |
request |
Union[google.cloud.tpu_v1.types.GetAcceleratorTypeRequest, dict]
The request object. Request for GetAcceleratorType. |
name |
str
Required. The resource name. This corresponds to the |
retry |
google.api_core.retry.Retry
Designation of what errors, if any, should be retried. |
timeout |
float
The timeout for this request. |
metadata |
Sequence[Tuple[str, str]]
Strings which should be sent along with the request as metadata. |
Returns | |
---|---|
Type | Description |
google.cloud.tpu_v1.types.AcceleratorType | A accelerator type that a Node can be configured with. |
get_mtls_endpoint_and_cert_source
get_mtls_endpoint_and_cert_source(
client_options: Optional[google.api_core.client_options.ClientOptions] = None,
)
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.
Parameter | |
---|---|
Name | Description |
client_options |
google.api_core.client_options.ClientOptions
Custom options for the client. Only the |
Exceptions | |
---|---|
Type | Description |
google.auth.exceptions.MutualTLSChannelError | If any errors happen. |
Returns | |
---|---|
Type | Description |
Tuple[str, Callable[[], Tuple[bytes, bytes]]] | returns the API endpoint and the client cert source to use. |
get_node
get_node(request: Optional[Union[google.cloud.tpu_v1.types.cloud_tpu.GetNodeRequest, dict]] = None, *, name: Optional[str] = None, retry: Union[google.api_core.retry.Retry, google.api_core.gapic_v1.method._MethodDefault] = <_MethodDefault._DEFAULT_VALUE: <object object>>, timeout: Optional[float] = None, metadata: Sequence[Tuple[str, str]] = ())
Gets the details of a node.
# This snippet has been automatically generated and should be regarded as a
# code template only.
# It will require modifications to work:
# - It may require correct/in-range values for request initialization.
# - It may require specifying regional endpoints when creating the service
# client as shown in:
# https://googleapis.dev/python/google-api-core/latest/client_options.html
from google.cloud import tpu_v1
def sample_get_node():
# Create a client
client = tpu_v1.TpuClient()
# Initialize request argument(s)
request = tpu_v1.GetNodeRequest(
name="name_value",
)
# Make the request
response = client.get_node(request=request)
# Handle the response
print(response)
Parameters | |
---|---|
Name | Description |
request |
Union[google.cloud.tpu_v1.types.GetNodeRequest, dict]
The request object. Request for GetNode. |
name |
str
Required. The resource name. This corresponds to the |
retry |
google.api_core.retry.Retry
Designation of what errors, if any, should be retried. |
timeout |
float
The timeout for this request. |
metadata |
Sequence[Tuple[str, str]]
Strings which should be sent along with the request as metadata. |
Returns | |
---|---|
Type | Description |
google.cloud.tpu_v1.types.Node | A TPU instance. |
get_tensor_flow_version
get_tensor_flow_version(request: Optional[Union[google.cloud.tpu_v1.types.cloud_tpu.GetTensorFlowVersionRequest, dict]] = None, *, name: Optional[str] = None, retry: Union[google.api_core.retry.Retry, google.api_core.gapic_v1.method._MethodDefault] = <_MethodDefault._DEFAULT_VALUE: <object object>>, timeout: Optional[float] = None, metadata: Sequence[Tuple[str, str]] = ())
Gets TensorFlow Version.
# This snippet has been automatically generated and should be regarded as a
# code template only.
# It will require modifications to work:
# - It may require correct/in-range values for request initialization.
# - It may require specifying regional endpoints when creating the service
# client as shown in:
# https://googleapis.dev/python/google-api-core/latest/client_options.html
from google.cloud import tpu_v1
def sample_get_tensor_flow_version():
# Create a client
client = tpu_v1.TpuClient()
# Initialize request argument(s)
request = tpu_v1.GetTensorFlowVersionRequest(
name="name_value",
)
# Make the request
response = client.get_tensor_flow_version(request=request)
# Handle the response
print(response)
Parameters | |
---|---|
Name | Description |
request |
Union[google.cloud.tpu_v1.types.GetTensorFlowVersionRequest, dict]
The request object. Request for GetTensorFlowVersion. |
name |
str
Required. The resource name. This corresponds to the |
retry |
google.api_core.retry.Retry
Designation of what errors, if any, should be retried. |
timeout |
float
The timeout for this request. |
metadata |
Sequence[Tuple[str, str]]
Strings which should be sent along with the request as metadata. |
Returns | |
---|---|
Type | Description |
google.cloud.tpu_v1.types.TensorFlowVersion | A tensorflow version that a Node can be configured with. |
list_accelerator_types
list_accelerator_types(request: Optional[Union[google.cloud.tpu_v1.types.cloud_tpu.ListAcceleratorTypesRequest, dict]] = None, *, parent: Optional[str] = None, retry: Union[google.api_core.retry.Retry, google.api_core.gapic_v1.method._MethodDefault] = <_MethodDefault._DEFAULT_VALUE: <object object>>, timeout: Optional[float] = None, metadata: Sequence[Tuple[str, str]] = ())
Lists accelerator types supported by this API.
# This snippet has been automatically generated and should be regarded as a
# code template only.
# It will require modifications to work:
# - It may require correct/in-range values for request initialization.
# - It may require specifying regional endpoints when creating the service
# client as shown in:
# https://googleapis.dev/python/google-api-core/latest/client_options.html
from google.cloud import tpu_v1
def sample_list_accelerator_types():
# Create a client
client = tpu_v1.TpuClient()
# Initialize request argument(s)
request = tpu_v1.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)
Parameters | |
---|---|
Name | Description |
request |
Union[google.cloud.tpu_v1.types.ListAcceleratorTypesRequest, dict]
The request object. Request for ListAcceleratorTypes. |
parent |
str
Required. The parent resource name. This corresponds to the |
retry |
google.api_core.retry.Retry
Designation of what errors, if any, should be retried. |
timeout |
float
The timeout for this request. |
metadata |
Sequence[Tuple[str, str]]
Strings which should be sent along with the request as metadata. |
Returns | |
---|---|
Type | Description |
google.cloud.tpu_v1.services.tpu.pagers.ListAcceleratorTypesPager | Response for ListAcceleratorTypes. Iterating over this object will yield results and resolve additional pages automatically. |
list_nodes
list_nodes(request: Optional[Union[google.cloud.tpu_v1.types.cloud_tpu.ListNodesRequest, dict]] = None, *, parent: Optional[str] = None, retry: Union[google.api_core.retry.Retry, google.api_core.gapic_v1.method._MethodDefault] = <_MethodDefault._DEFAULT_VALUE: <object object>>, timeout: Optional[float] = None, metadata: Sequence[Tuple[str, str]] = ())
Lists nodes.
# This snippet has been automatically generated and should be regarded as a
# code template only.
# It will require modifications to work:
# - It may require correct/in-range values for request initialization.
# - It may require specifying regional endpoints when creating the service
# client as shown in:
# https://googleapis.dev/python/google-api-core/latest/client_options.html
from google.cloud import tpu_v1
def sample_list_nodes():
# Create a client
client = tpu_v1.TpuClient()
# Initialize request argument(s)
request = tpu_v1.ListNodesRequest(
parent="parent_value",
)
# Make the request
page_result = client.list_nodes(request=request)
# Handle the response
for response in page_result:
print(response)
Parameters | |
---|---|
Name | Description |
request |
Union[google.cloud.tpu_v1.types.ListNodesRequest, dict]
The request object. Request for ListNodes. |
parent |
str
Required. The parent resource name. This corresponds to the |
retry |
google.api_core.retry.Retry
Designation of what errors, if any, should be retried. |
timeout |
float
The timeout for this request. |
metadata |
Sequence[Tuple[str, str]]
Strings which should be sent along with the request as metadata. |
Returns | |
---|---|
Type | Description |
google.cloud.tpu_v1.services.tpu.pagers.ListNodesPager | Response for ListNodes. Iterating over this object will yield results and resolve additional pages automatically. |
list_tensor_flow_versions
list_tensor_flow_versions(request: Optional[Union[google.cloud.tpu_v1.types.cloud_tpu.ListTensorFlowVersionsRequest, dict]] = None, *, parent: Optional[str] = None, retry: Union[google.api_core.retry.Retry, google.api_core.gapic_v1.method._MethodDefault] = <_MethodDefault._DEFAULT_VALUE: <object object>>, timeout: Optional[float] = None, metadata: Sequence[Tuple[str, str]] = ())
List TensorFlow versions supported by this API.
# This snippet has been automatically generated and should be regarded as a
# code template only.
# It will require modifications to work:
# - It may require correct/in-range values for request initialization.
# - It may require specifying regional endpoints when creating the service
# client as shown in:
# https://googleapis.dev/python/google-api-core/latest/client_options.html
from google.cloud import tpu_v1
def sample_list_tensor_flow_versions():
# Create a client
client = tpu_v1.TpuClient()
# Initialize request argument(s)
request = tpu_v1.ListTensorFlowVersionsRequest(
parent="parent_value",
)
# Make the request
page_result = client.list_tensor_flow_versions(request=request)
# Handle the response
for response in page_result:
print(response)
Parameters | |
---|---|
Name | Description |
request |
Union[google.cloud.tpu_v1.types.ListTensorFlowVersionsRequest, dict]
The request object. Request for ListTensorFlowVersions. |
parent |
str
Required. The parent resource name. This corresponds to the |
retry |
google.api_core.retry.Retry
Designation of what errors, if any, should be retried. |
timeout |
float
The timeout for this request. |
metadata |
Sequence[Tuple[str, str]]
Strings which should be sent along with the request as metadata. |
Returns | |
---|---|
Type | Description |
google.cloud.tpu_v1.services.tpu.pagers.ListTensorFlowVersionsPager | Response for ListTensorFlowVersions. Iterating over this object will yield results and resolve additional pages automatically. |
node_path
node_path(project: str, location: str, node: str)
Returns a fully-qualified node string.
parse_accelerator_type_path
parse_accelerator_type_path(path: str)
Parses a accelerator_type path into its component segments.
parse_common_billing_account_path
parse_common_billing_account_path(path: str)
Parse a billing_account path into its component segments.
parse_common_folder_path
parse_common_folder_path(path: str)
Parse a folder path into its component segments.
parse_common_location_path
parse_common_location_path(path: str)
Parse a location path into its component segments.
parse_common_organization_path
parse_common_organization_path(path: str)
Parse a organization path into its component segments.
parse_common_project_path
parse_common_project_path(path: str)
Parse a project path into its component segments.
parse_node_path
parse_node_path(path: str)
Parses a node path into its component segments.
parse_tensor_flow_version_path
parse_tensor_flow_version_path(path: str)
Parses a tensor_flow_version path into its component segments.
reimage_node
reimage_node(request: Optional[Union[google.cloud.tpu_v1.types.cloud_tpu.ReimageNodeRequest, dict]] = None, *, retry: Union[google.api_core.retry.Retry, google.api_core.gapic_v1.method._MethodDefault] = <_MethodDefault._DEFAULT_VALUE: <object object>>, timeout: Optional[float] = None, metadata: Sequence[Tuple[str, str]] = ())
Reimages a node's OS.
# This snippet has been automatically generated and should be regarded as a
# code template only.
# It will require modifications to work:
# - It may require correct/in-range values for request initialization.
# - It may require specifying regional endpoints when creating the service
# client as shown in:
# https://googleapis.dev/python/google-api-core/latest/client_options.html
from google.cloud import tpu_v1
def sample_reimage_node():
# Create a client
client = tpu_v1.TpuClient()
# Initialize request argument(s)
request = tpu_v1.ReimageNodeRequest(
)
# Make the request
operation = client.reimage_node(request=request)
print("Waiting for operation to complete...")
response = operation.result()
# Handle the response
print(response)
Parameters | |
---|---|
Name | Description |
request |
Union[google.cloud.tpu_v1.types.ReimageNodeRequest, dict]
The request object. Request for ReimageNode. |
retry |
google.api_core.retry.Retry
Designation of what errors, if any, should be retried. |
timeout |
float
The timeout for this request. |
metadata |
Sequence[Tuple[str, str]]
Strings which should be sent along with the request as metadata. |
Returns | |
---|---|
Type | Description |
google.api_core.operation.Operation | An object representing a long-running operation. The result type for the operation will be Node A TPU instance. |
start_node
start_node(request: Optional[Union[google.cloud.tpu_v1.types.cloud_tpu.StartNodeRequest, dict]] = None, *, retry: Union[google.api_core.retry.Retry, google.api_core.gapic_v1.method._MethodDefault] = <_MethodDefault._DEFAULT_VALUE: <object object>>, timeout: Optional[float] = None, metadata: Sequence[Tuple[str, str]] = ())
Starts a node.
# This snippet has been automatically generated and should be regarded as a
# code template only.
# It will require modifications to work:
# - It may require correct/in-range values for request initialization.
# - It may require specifying regional endpoints when creating the service
# client as shown in:
# https://googleapis.dev/python/google-api-core/latest/client_options.html
from google.cloud import tpu_v1
def sample_start_node():
# Create a client
client = tpu_v1.TpuClient()
# Initialize request argument(s)
request = tpu_v1.StartNodeRequest(
)
# Make the request
operation = client.start_node(request=request)
print("Waiting for operation to complete...")
response = operation.result()
# Handle the response
print(response)
Parameters | |
---|---|
Name | Description |
request |
Union[google.cloud.tpu_v1.types.StartNodeRequest, dict]
The request object. Request for StartNode. |
retry |
google.api_core.retry.Retry
Designation of what errors, if any, should be retried. |
timeout |
float
The timeout for this request. |
metadata |
Sequence[Tuple[str, str]]
Strings which should be sent along with the request as metadata. |
Returns | |
---|---|
Type | Description |
google.api_core.operation.Operation | An object representing a long-running operation. The result type for the operation will be Node A TPU instance. |
stop_node
stop_node(request: Optional[Union[google.cloud.tpu_v1.types.cloud_tpu.StopNodeRequest, dict]] = None, *, retry: Union[google.api_core.retry.Retry, google.api_core.gapic_v1.method._MethodDefault] = <_MethodDefault._DEFAULT_VALUE: <object object>>, timeout: Optional[float] = None, metadata: Sequence[Tuple[str, str]] = ())
Stops a node.
# This snippet has been automatically generated and should be regarded as a
# code template only.
# It will require modifications to work:
# - It may require correct/in-range values for request initialization.
# - It may require specifying regional endpoints when creating the service
# client as shown in:
# https://googleapis.dev/python/google-api-core/latest/client_options.html
from google.cloud import tpu_v1
def sample_stop_node():
# Create a client
client = tpu_v1.TpuClient()
# Initialize request argument(s)
request = tpu_v1.StopNodeRequest(
)
# Make the request
operation = client.stop_node(request=request)
print("Waiting for operation to complete...")
response = operation.result()
# Handle the response
print(response)
Parameters | |
---|---|
Name | Description |
request |
Union[google.cloud.tpu_v1.types.StopNodeRequest, dict]
The request object. Request for StopNode. |
retry |
google.api_core.retry.Retry
Designation of what errors, if any, should be retried. |
timeout |
float
The timeout for this request. |
metadata |
Sequence[Tuple[str, str]]
Strings which should be sent along with the request as metadata. |
Returns | |
---|---|
Type | Description |
google.api_core.operation.Operation | An object representing a long-running operation. The result type for the operation will be Node A TPU instance. |
tensor_flow_version_path
tensor_flow_version_path(project: str, location: str, tensor_flow_version: str)
Returns a fully-qualified tensor_flow_version string.