Skip to content

Latest commit

 

History

History
25 lines (17 loc) · 878 Bytes

first_run.md

File metadata and controls

25 lines (17 loc) · 878 Bytes

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 (v4-8).
  2. Clone MaxDiffusion in your TPU VM.
  3. Within the root directory of the MaxDiffusion git repo, install dependencies by running:
If you are running on TPU:
bash setup.sh MODE=stable DEVICE=tpu

If you are running on GPU:
bash setup.sh MODE=stable DEVICE=gpu

Getting Starting: Multihost development

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