torchx

Torchx

Torchx ist ein universeller Job -Launcher für Pytorch -Anwendungen. Torchx ist so konzipiert, dass sie eine schnelle Iterationszeit für Schulungen/Forschung und Unterstützung für ML -Pipelines der E2E -Produktion haben, wenn Sie bereit sind.

Torchx unterstützt derzeit:

• Kubernetes (Eks, GKE, AKs usw.)
• Schläge
• AWS -Stapel
• Docker
• Lokal
• Strahl (Prototyp)
• GCP -Stapel (Prototyp)

Benötigen Sie einen nicht aufgeführten Scheduler? Lass es uns wissen!

QuickStart

Siehe den QuickStart Guide.

Dokumentation

• Stabile Dokumentation
• Nachtdokumentation

Anforderungen

Torchx:

• Python3 (3,8+)
• Pytorch
• Optional: Docker (benötigt für Docker -basierte Planer)

Bestimmte Scheduler benötigen möglicherweise planenspezifische Anforderungen. Informationen finden Sie in der Installation.

Installation

Stabil

 # install torchx sdk and CLI -- minimum dependencies
pip install torchx

# install torchx sdk and CLI -- all dependencies
pip install " torchx[dev] "

# install torchx kubeflow pipelines (kfp) support
pip install " torchx[kfp] "

# install torchx Kubernetes / Volcano support
pip install " torchx[kubernetes] "

# install torchx Ray support
pip install " torchx[ray] "

# install torchx GCP Batch support
pip install " torchx[gcp_batch] "

Nacht

 # install torchx sdk and CLI
pip install torchx-nightly[dev]

Quelle

 # install torchx sdk and CLI from source
$ pip install -e git+https://github.com/pytorch/torchx.git#egg=torchx

# install extra dependencies
$ pip install -e git+https://github.com/pytorch/torchx.git#egg=torchx[dev]

Docker

Torchx bietet einen Docker -Container für die Verwendung als Teil einer Torchx -Rolle.

Siehe: https://github.com/pytorch/torchx/pkgs/container/torchx

Beitragen

Wir begrüßen PRS! Siehe die beitragende Datei.

Lizenz

Torchx ist BSD lizenziert, wie in der Lizenzdatei zu finden.