torchx

Torchx

Torchx est un lanceur d'emplois universel pour les applications Pytorch. Torchx est conçu pour avoir un temps d'itération rapide pour la formation / la recherche et le soutien pour les pipelines ML de production E2E lorsque vous êtes prêt.

Torchx prend actuellement en charge:

• Kubernetes (EKS, GKE, AKS, etc.)
• Slurm
• Lot AWS
• Docker
• Locale
• Ray (prototype)
• Lot GCP (prototype)

Besoin d'un planificateur non répertorié? Faites-le nous savoir!

Start

Voir le guide QuickStart.

Documentation

• Documentation stable
• Documentation nocturne

Exigences

Torchx:

• Python3 (3.8+)
• Pytorch
• Facultatif: Docker (nécessaire pour les planificateurs basés sur Docker)

Certains planificateurs peuvent exiger des exigences spécifiques à l'ordonnanceur. Voir l'installation pour plus d'informations.

Installation

Écurie

 # 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] "

Chaque nuit

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

Source

 # 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 fournit un conteneur Docker à utiliser dans le cadre d'un rôle Torchx.

Voir: https://github.com/pytorch/torchx/pkgs/containener/torchx

Contributif

Nous accueillons les PRS! Voir le fichier contributif.

Licence

Torchx est sous licence BSD, comme le montre le fichier de licence.