ray.util.tpu.get_tpu_version_from_type

ray.util.tpu.get_tpu_version_from_type(accelerator_type: str) → str

Extracts the version from the accelerator type.

Parameters:

accelerator_type – The full accelerator type string (e.g. “TPU-V6E”).

Returns:

The version string (e.g. “v6e”).

Raises:

ValueError – If the accelerator type is invalid.

PublicAPI (alpha): This API is in alpha and may change before becoming stable.