Skip to content

Commit 76f84a1

Browse files
authored
Updating getting started guide to point users to use a more recent ve… (#194)
* Updating getting started guide to point users to use a more recent version of TPU (v6 instead of v4, which was hard to find), and simplifying content * Update first_run.md
1 parent 9b07874 commit 76f84a1

1 file changed

Lines changed: 9 additions & 5 deletions

File tree

docs/getting_started/first_run.md

Lines changed: 9 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -8,15 +8,19 @@ We recommend starting with a single host first and then moving to multihost.
88
Local development is a convenient way to run MaxDiffusion on a single host. It doesn't scale to
99
multiple hosts.
1010

11-
1. [Create and SSH to a single-host TPU (v4-8). ](https://cloud.google.com/tpu/docs/users-guide-tpu-vm#creating_a_cloud_tpu_vm_with_gcloud)
11+
1. [Create and SSH to a single-host TPU (v6-8). ](https://cloud.google.com/tpu/docs/users-guide-tpu-vm#creating_a_cloud_tpu_vm_with_gcloud)
12+
* You can find here [here](https://cloud.google.com/tpu/docs/regions-zones) the list of zones that support the v6(Trillium) TPUs
13+
* We recommend using the base VM image "v2-alpha-tpuv6e", which meets the version requirements: Ubuntu Version 22.04, Python 3.10 and Tensorflow >= 2.12.0
14+
1215
1. Clone MaxDiffusion in your TPU VM.
16+
```bash
17+
git clone https://github.com/AI-Hypercomputer/maxdiffusion.git
18+
cd maxdiffusion
19+
```
20+
1321
1. Within the root directory of the MaxDiffusion `git` repo, install dependencies by running:
1422
```bash
15-
If you are running on TPU:
1623
bash setup.sh MODE=stable DEVICE=tpu
17-
18-
If you are running on GPU:
19-
bash setup.sh MODE=stable DEVICE=gpu
2024
```
2125

2226
## Getting Starting: Multihost development

0 commit comments

Comments
 (0)