跳到正文
Google Developers Blog·· 2026-07-20AI 评分56

Ray 2.55 正式支持 Google Cloud TPU,通过 KubeRay 自动编排多主机切片

Run Ray on TPU, Part 1: The foundations

AI 导读

Ray 2.55 首次为 Google Cloud TPU 提供一等支持,开发者可通过 Ray 任务与 Actor API 在 TPU 上运行分布式 Python 负载。

正文 · 原文

Ray 2.55 introduces official, first-class support for Google Cloud TPUs, enabling developers to run distributed Python workloads on Google's accelerators using the familiar Ray task-and-actor APIs. To handle the strict networking requirement of keeping multi-host TPU "slices" together over their Inter-Chip Interconnect (ICI), the KubeRay Operator on GKE automatically provisions and labels the underlying hardware layout. Ray Core utilizes these labels via its slice_placement_group() primitive to atomically reserve complete slices, allowing developers to deploy jobs through KubeRay, Ray Train, or Ray Serve simply by declaring a hardware topology (like "4x4") without writing custom placement code.

来源:Google Developers Blog · developers.googleblog.com