torchx

Antorchx

Torchx es un lanzador de trabajo universal para aplicaciones de Pytorch. Torchx está diseñado para tener un tiempo de iteración rápido para capacitación/investigación y apoyo para tuberías de producción de E2E ML cuando esté listo.

Torchx actualmente es compatible:

• Kubernetes (EKS, GKE, AKS, etc.)
• Slurm
• Lote de AWS
• Estibador
• Local
• Ray (prototipo)
• GCP Batch (prototipo)

¿Necesita un programador que no esté en la lista? ¡Háganos saber!

Inicio rápido

Consulte la guía QuickStart.

Documentación

• Documentación estable
• Documentación nocturna

Requisitos

Torchx:

• Python3 (3.8+)
• Pytorch
• Opcional: Docker (necesario para programadores basados en Docker)

Ciertos programadores pueden requerir requisitos específicos del planificador. Consulte la instalación para obtener información.

Instalación

Estable

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

Nocturno

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

Fuente

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

Estibador

TORCHX proporciona un contenedor Docker para usar como parte de un rol de Torchx.

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

Que contribuye

¡Damos la bienvenida a PRS! Ver el archivo contribuyente.

Licencia

Torchx tiene licencia BSD, como se encuentra en el archivo de licencia.