TPU 操作系统映像
使用 Compute Engine 创建 TPU 虚拟机时，您可以使用 Google 管理的公共操作系统 (OS) 映像或自定义映像。 Google 管理的映像针对 TPU 性能进行了优化，并包含必要的驱动程序和代理。默认情况下，所有 Google Cloud 项目都可以访问这些公共操作系统映像。
Google 管理的操作系统映像
Google 提供基于 Ubuntu 的操作系统映像，这些映像预配置了 TPU 运行时、驱动程序和代理，以便您的工作负载能够正常运行。
下表列出了每个受支持的 TPU 版本的 Google 管理的操作系统映像：
| TPU 版本 | 操作系统版本 | 内核版本 | 映像项目 | 映像系列 |
|---|---|---|---|---|
| TPU7x (Ironwood) | Ubuntu 24.04 LTS |
6.17 | ubuntu-os-accelerator-images |
ubuntu-accel-2404-amd64-tpu-tpu7x |
| v6e (Trillium)、v5p | Ubuntu 22.04 LTS |
6.8 | ubuntu-os-accelerator-images |
ubuntu-accel-2204-amd64-tpu-v5e-v5p-v6e |
自定义操作系统映像
借助 Compute Engine，您可以为 TPU 虚拟机使用自定义操作系统映像。自定义映像适用于以下情况：
- 预加载自定义软件和库，以缩短虚拟机启动时间。
- 使用 Google 管理的列表中未包含的操作系统发行版或版本。
- 应用自定义内核修改或安全配置。
责任共担模型
使用自定义操作系统映像时，您和 Google 需遵循责任共担模型，如下表所示：
| 功能 | Google 的责任 | 用户的责任 |
|---|---|---|
| TPU 代理和驱动程序 | 发布修改并发布代理 | 安装和配置代理和驱动程序 |
| 兼容性测试 | 不适用 | 执行兼容性和性能测试 |
| 操作系统维护 | 不适用 | 管理操作系统更新和许可 |
如需详细了解如何构建和使用您自己的 Ubuntu 映像，请参阅使用 自定义操作系统映像。