Skip to content

first_run.md

Latest commit

 

History

History
29 lines (20 loc) · 1.14 KB

first_run.md

File metadata and controls

29 lines (20 loc) · 1.14 KB

Getting Started

We recommend starting with a single host first and then moving to multihost.

Getting Started: Local Development for single host

Running on Cloud TPUs

Local development is a convenient way to run MaxDiffusion on a single host. It doesn't scale to multiple hosts.

  1. Create and SSH to a single-host TPU (v6-8).
  • You can find here here the list of zones that support the v6(Trillium) TPUs
  • 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
  1. Clone MaxDiffusion in your TPU VM.
git clone https://github.com/AI-Hypercomputer/maxdiffusion.git
cd maxdiffusion
  1. Within the root directory of the MaxDiffusion git repo, install dependencies by running:
bash setup.sh MODE=stable DEVICE=tpu

Getting Starting: Multihost development

[GKE, recommended] Running MaxDiffusion with xpk - Quick Experimentation and Production support