本页面上的部分或全部信息可能不适用于 Cloud de Confiance by S3NS。如需了解详情，请参阅与 Google Cloud 的区别。

Google uses AI technology to translate content into your preferred language. AI translations can contain errors.

创建 TPU 实例

本文档介绍了如何创建 TPU v6e (Trillium) 虚拟机 (VM) 实例、连接到该实例以及在该实例上运行计算。您在本快速入门中创建的 TPU 实例使用按需付费，这是默认的使用选项。

准备工作

安装 Google Cloud CLI。

注意：如果您之前安装了 gcloud CLI，请确保通过运行 gcloud components update 来获得最新版本。
配置 gcloud CLI 以使用您的联合身份。

如需了解详情，请参阅使用联合身份登录 gcloud CLI。
如需初始化 gcloud CLI，请运行以下命令：
```
gcloud init
```
创建或选择 Cloud de Confiance 项目。
选择或创建项目所需角色
- 选择项目：选择项目不需要特定的 IAM 角色，您可以选择已获授角色的任何项目。
- 创建项目：如需创建项目，您需要 Project Creator 角色 (roles/resourcemanager.projectCreator)，该角色包含 resourcemanager.projects.create 权限。了解如何授予角色。
注意：如果您不打算保留在此过程中创建的资源，请创建新的项目，而不要选择现有的项目。完成本教程介绍的步骤后，您可以删除所创建的项目，并移除与该项目关联的所有资源。
- 创建 Cloud de Confiance 项目：
```
gcloud projects create PROJECT_ID
```
  将 PROJECT_ID 替换为您要创建的 Cloud de Confiance 项目的名称。
- 选择您创建的 Cloud de Confiance 项目：
```
gcloud config set project PROJECT_ID
```
  将 PROJECT_ID 替换为您的 Cloud de Confiance 项目名称。
如果您要使用现有项目来完成本指南，请验证您是否拥有完成本指南所需的权限。如果您创建了新项目，则您已拥有所需的权限。
验证是否已为您的 Cloud de Confiance 项目启用结算功能。
启用 Compute Engine API：
启用 API 所需的角色
如需启用 API，您需要拥有 serviceusage.services.enable 权限。如果您创建了项目，则您可能已通过 Owner 角色 (roles/owner) 拥有此权限。否则，您可以通过 Service Usage Admin 角色 (roles/serviceusage.serviceUsageAdmin) 获得此权限。了解如何授予角色。
```
gcloud services enable compute.googleapis.com
```

所需的角色

如需获得创建 TPU 实例、使用 SSH 连接到该实例以及运行命令所需的权限，请让您的管理员为您授予项目的以下 IAM 角色：

Compute Instance Admin (v1) (roles/compute.instanceAdmin.v1)
Service Account User (roles/iam.serviceAccountUser)
Service Usage Admin (roles/serviceusage.serviceUsageAdmin)

如需详细了解如何授予角色，请参阅管理对项目、文件夹和组织的访问权限。

您也可以通过自定义角色或其他预定义角色来获取所需的权限。

创建 TPU v6e (Trillium) 实例

在本地终端中或使用 Cloud Shell运行以下命令。

如需创建 TPU v6e 实例，请运行以下 gcloud compute instances create 命令：

gcloud compute instances create quickstart-tpu \
    --machine-type=ct6e-standard-4t \
    --zone=europe-west4-a \
    --image-project=ubuntu-os-accelerator-images \
    --image-family=ubuntu-accel-2204-amd64-tpu-v5e-v5p-v6e \
    --maintenance-policy=TERMINATE

（可选）使用 gcloud compute instances describe 命令验证实例是否正在运行：
```
gcloud compute instances describe quickstart-tpu --zone=europe-west4-a
```
在输出中，检查 status 字段以验证实例是否正在运行或仍在等待。

连接到 TPU 实例

您可以使用 Cloud de Confiance 控制台或 gcloud CLI 连接到 TPU 实例。

如需使用 SSH 连接到虚拟机，请运行以下 gcloud compute ssh 命令：

gcloud compute ssh quickstart-tpu --zone=europe-west4-a

安装 JAX

在 TPU 实例上运行以下命令以安装 JAX 框架：

pip install jax[tpu] -f https://storage.googleapis.com/jax-releases/libtpu_releases.html

运行系统检查

在 TPU 实例上运行以下命令，验证 JAX 是否可以访问 TPU 且可以运行基本操作：

启动 Python 3 解释器：
```
python3
```
导入 JAX 模块：
```
>>> import jax
```
显示可用的 TPU 核心数：
```
>>> jax.device_count()
```
输出会显示 TPU 核心数。核心数取决于您使用的 TPU 版本。如需了解详情，请参阅 TPU 版本。

执行计算

运行以下命令以将两个数字相加：
```
>>> jax.numpy.add(1, 1)
```
输出应类似如下所示：
```
>>> Array(2, dtype=int32, weak_type=True)
```
退出 Python 解释器：
```
>>> 
exit()
```

清理

为避免因本页面中使用的资源导致您的 Cloud de Confiance 账号产生费用，请删除包含这些资源的 Cloud de Confiance 项目。

或者，如果您想保留项目，则可以使用 gcloud compute instances delete 命令仅删除虚拟机：

gcloud compute instances delete quickstart-tpu --zone=europe-west4-a