You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: examples/gke-tpu-7x/README.md
+7-7Lines changed: 7 additions & 7 deletions
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -10,7 +10,7 @@ The blueprint follows Google Cloud best practices, including:
10
10
11
11
> NOTE: This guide provides examples for deploying a small, single-node slice (2x2x1), but the parameters in the deployment file can be easily changed to provision larger, multi-node slices.
12
12
13
-
This guide also includes a `tpu-7x-job.yaml` that creates a Kubernetes Pod and runs a simple command to check for available TPU chips.
13
+
This guide also includes a `gke-tpu-7x-job.yaml` that creates a Kubernetes Pod and runs a simple command to check for available TPU chips.
14
14
15
15
## Key parameters when working with TPUs
16
16
@@ -126,8 +126,8 @@ The process is nearly identical to the basic deployment.
3. After deployment, the blueprint will output instructions for running a fio benchmark job. This job serves as a validation test to confirm that the GCS mounts are working correctly for both reading and writing. Follow the printed instructions to run the test.
@@ -153,7 +153,7 @@ This blueprint supports [Kueue](https://kueue.sigs.k8s.io/), a kubernetes-native
153
153
154
154
## Run the sample job
155
155
156
-
The `tpu-7x-job.yaml` file creates a Pod resource in Kubernetes. The workload installs JAX and a specific `libtpu` library, and then returns the number of TPU chips it can detect.
156
+
The `gke-tpu-7x-job.yaml` file creates a Pod resource in Kubernetes. The workload installs JAX and a specific `libtpu` library, and then returns the number of TPU chips it can detect.
157
157
158
158
1. Connect to your cluster:
159
159
@@ -165,7 +165,7 @@ The `tpu-7x-job.yaml` file creates a Pod resource in Kubernetes. The workload in
165
165
166
166
Replace `DEPLOYMENT_NAME`, `REGION`, and `PROJECT_ID` with the ones used in your `gke-tpu-7x-deployment.yaml` file.
167
167
2. Update the `nodeSelector`in the job file.
168
-
Open the `examples/gke-tpu-7x/tpu-7x-job.yaml` file. Ensure the `nodeSelector` values match the accelerator label and topology used in your deployment file.
168
+
Open the `examples/gke-tpu-7x/gke-tpu-7x-job.yaml` file. Ensure the `nodeSelector` values match the accelerator label and topology used in your deployment file.
169
169
>*You can find the correct labels for your cluster by running `kubectl get nodes --show-labels`*
170
170
171
171
For the example deployment, the values are:
@@ -179,7 +179,7 @@ The `tpu-7x-job.yaml` file creates a Pod resource in Kubernetes. The workload in
179
179
3. Update Resource Request:
180
180
181
181
```yaml
182
-
# In tpu-7x-job.yaml, inside the spec container
182
+
# In gke-tpu-7x-job.yaml, inside the spec container
183
183
resources:
184
184
requests:
185
185
google.com/tpu: <CHIPS_PER_NODE># e.g., 4 for tpu7x-standard-4t
@@ -191,7 +191,7 @@ The `tpu-7x-job.yaml` file creates a Pod resource in Kubernetes. The workload in
191
191
The sample job file already contains the correct node selectors for the example deployment.
0 commit comments