TPU 操作系统映像
使用 Compute Engine 创建 TPU 虚拟机时,您可以使用 Google 管理的公共操作系统 (OS) 映像或自定义映像。Google 管理的映像针对 TPU 性能进行了优化,并包含必要的驱动程序和代理。默认情况下,所有 Google Cloud 项目都可以访问这些公共操作系统映像。
Google 管理的操作系统映像
Google 提供基于 Ubuntu 的操作系统映像,这些映像预配置了工作负载运行所需的 TPU 运行时、驱动程序和代理。
下表列出了每个受支持的 TPU 版本的 Google 管理的操作系统映像:
| TPU 版本 | 操作系统版本 | 内核版本 | 映像项目 | 映像系列 |
|---|---|---|---|---|
| TPU7x (Ironwood) | Ubuntu 24.04 LTS |
6.8 | 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 管理的列表中的操作系统发行版或版本。
- 应用自定义内核修改或安全配置。
TPU7x 虚拟机要求客户机操作系统 Linux 内核版本不低于 6.8。
责任共担模型
使用自定义操作系统映像遵循您与 Google 之间的责任共担模型,如下表所示:
| 功能 | Google 责任 | 用户责任 |
|---|---|---|
| TPU 代理和驱动程序 | 发布修改和发布智能体 | 安装和配置代理及驱动程序 |
| 兼容性测试 | 不适用 | 执行兼容性和性能测试 |
| 操作系统维护 | 不适用 | 管理操作系统更新和许可 |
如需详细了解如何构建和使用您自己的 Ubuntu 映像,请参阅使用自定义操作系统映像。