跳转到主要内容

TorchX SDK和组件

项目描述

PyPI License Tests Lint codecov

TorchX

TorchX是一个通用的PyTorch应用程序作业启动器。TorchX旨在具有快速迭代时间以进行训练/研究,并在您准备好时支持端到端生产机器学习管道。

TorchX目前支持

  • Kubernetes (EKS, GKE, AKS, 等)
  • Slurm
  • AWS Batch
  • Docker
  • 本地
  • Ray (原型)
  • GCP Batch (原型)

需要未列出的调度器?告诉我们!

快速入门

查看快速入门指南

文档

要求

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!请参阅CONTRIBUTING文件。

许可证

TorchX采用BSD许可,如LICENSE文件所示。

项目详情


发布历史 发布通知 | RSS源

下载文件

下载适用于您平台的文件。如果您不确定选择哪个,请了解有关安装包的更多信息。

源代码分发

此版本没有可用的源代码分发文件。请参阅生成分发存档的教程

构建分发

torchx_nightly-2024.10.4-py3-none-any.whl (259.7 kB 查看哈希值)

上传时间 Python 3

由以下支持