torchx

Torchx

Torchx é um lançador de empregos universal para aplicativos Pytorch. O Torchx foi projetado para ter tempo de iteração rápida para treinamento/pesquisa e suporte para os pipelines da E2E Production ML quando estiver pronto.

Torchx atualmente suporta:

• Kubernetes (EKS, GKE, AKS, etc)
• Slurm
• Lote AWS
• Docker
• Local
• Ray (protótipo)
• GCP Lote (protótipo)

Precisa de um agendador não listado? Deixe -nos saber!

Investir rápido

Veja o Guia do QuickStart.

Documentação

• Documentação estável
• Documentação noturna

Requisitos

Torchx:

• Python3 (3,8+)
• Pytorch
• Opcional: Docker (necessário para agendadores baseados em docker)

Certos agendadores podem exigir requisitos específicos do agendador. Consulte a instalação para obter informações.

Instalação

Estável

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

Todas as noites

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

Fonte

 # 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

O Torchx fornece um contêiner do Docker para usar como parte de uma função de Torchx.

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

Contribuindo

Congratulamo -nos com o PRS! Veja o arquivo contribuinte.

Licença

O Torchx é licenciado em BSD, conforme encontrado no arquivo de licença.