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_nightly-2024.10.4-py3-none-any.whl (259.7 kB 查看哈希值)