torchx

Torchx

Torchx是Pytorch应用程序的通用启动器。 Torchx旨在有快速的迭代时间在准备就绪后对E2E生产ML管道进行培训/研究和支持。

Torchx当前支持：

• kubernetes（eks，gke，aks等）
• sl
• AWS批次
• Docker
• 当地的
• 射线（原型）
• GCP批次（原型）

需要未列出的调度程序吗？让我们知道！

Quickstart

请参阅《快速启动指南》。

文档

• 稳定的文档
• 夜间文档

要求

TORCHX：

• Python3（3.8+）
• Pytorch
• 可选：Docker（基于Docker的调度程序需要）

某些调度程序可能需要调度程序特定要求。有关信息，请参见安装。

安装

稳定的

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

每晚

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

来源

 # 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提供了一个docker容器，用于用作TORCHX角色的一部分。

请参阅：https：//github.com/pytorch/torchx/pkgs/container/torchx

贡献

我们欢迎PR！请参阅贡献文件。

执照

如许可证文件中所示，TORCHX已获得BSD许可。