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許可。