easydel.scripts.install_on_hosts

Contents

easydel.scripts.install_on_hosts#

Install EasyDel and dependencies on Ray TPU pods.

This script automates the installation of EasyDel and its dependencies across all hosts in a TPU pod. It uses Ray for distributed execution and TPUExecutor for TPU-specific orchestration.

Usage:

python -m easydel.scripts.install_on_hosts –tpu-type <TPU_TYPE> –source <pypi|github>

Options:
--tpu-type

TPU pod slice type (e.g. v4-16, v3-8)

--source

Installation source: ‘pypi’ for PyPI package or ‘github’ for latest from GitHub

–num-tpu-hosts Override default host count for TPU type

Example

python -m easydel.scripts.install_on_hosts –tpu-type v4-16 –source github

easydel.scripts.install_on_hosts.main()[source]#