diff --git a/applications/ray/TPU_guide.md b/applications/ray/TPU_guide.md index 29fa5b9e8..62f92451e 100644 --- a/applications/ray/TPU_guide.md +++ b/applications/ray/TPU_guide.md @@ -38,7 +38,7 @@ accelerator_type = "nvidia-tesla-t4" ### Installing the TPU Initialization Webhook -The TPU Initialization Webhook automatically injects the `TPU_WORKER_ID`, `TPU_NAME`, and `TPU_WORKER_HOSTNAMES` environment variables necessary for multi-host TPU clusters. The webhook needs to be installed once per GKE cluster. The instructions can be found [here](https://github.com/GoogleCloudPlatform/ai-on-gke/blob/main/applications/ray/kuberay-tpu-webhook). +The TPU Initialization Webhook automatically injects the `TPU_WORKER_ID`, `TPU_NAME`, and `TPU_WORKER_HOSTNAMES` environment variables necessary for multi-host TPU clusters. The webhook needs to be installed once per GKE cluster and requires a Kuberay Operator running v1.1 and GKE cluster version of 1.28+. The instructions can be found [here](https://github.com/GoogleCloudPlatform/ai-on-gke/blob/main/applications/ray/kuberay-tpu-webhook). ### Creating the Kuberay Cluster