TorchX SDK和组件
项目描述
TorchX
TorchX是PyTorch应用程序的通用作业启动器。TorchX旨在具有快速的迭代时间进行训练/研究，并在您准备好时支持端到端生产机器学习管道。
TorchX目前支持
- Kubernetes (EKS, GKE, AKS等)
- Slurm
- AWS Batch
- Docker
- 本地
- Ray（原型）
- GCP Batch（原型）
需要未列出的调度器？ 告诉我们！
快速入门
查看快速入门指南。
文档
需求
torchx
某些调度器可能需要特定于调度器的需求。有关安装信息，请参阅安装说明。
安装
稳定
# 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！请参阅CONTRIBUTING文件。
许可证
TorchX遵循BSD许可证，如LICENSE文件所示。
下载文件
下载您平台的文件。如果您不确定选择哪个，请了解有关安装包的更多信息。
源代码发行版
torchx-0.7.0.tar.gz （196.1 KB 查看哈希值）
构建发行版
torchx-0.7.0-py3-none-any.whl （256.1 KB 查看哈希值）