Skip to content

README: update readme docs and requirement links #407

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 1 commit into from
Closed
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
32 changes: 23 additions & 9 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -12,25 +12,30 @@ TorchX is a universal job launcher for PyTorch applications.
TorchX is designed to have fast iteration time for training/research and support
for E2E production ML pipelines when you're ready.

For the latest documentation, please refer to our [website](https://pytorch.org/torchx).

## Quickstart

See the [quickstart guide](https://pytorch.org/torchx/latest/quickstart.html).

## Documentation

* [Stable Documentation](https://pytorch.org/torchx/latest/)
* [Nightly Documentation](https://pytorch.org/torchx/main/)

## Requirements
TorchX SDK (torchx):

torchx:

* python3 (3.8+)
* torch
* [PyTorch](https://pytorch.org/get-started/locally/)
* optional: [Docker](https://docs.docker.com/get-docker/) (needed for docker based schedulers)

TorchX Kubeflow Pipelines Support (torchx-kfp):
* torchx
* kfp
Certain schedulers may require scheduler specific requirements. See installation
for info.

## Installation

### Release
### Stable

```bash
# install torchx sdk and CLI -- minimum dependencies
Expand All @@ -44,13 +49,16 @@ pip install "torchx[kfp]"

# install torchx Kubernetes / Volcano support
pip install "torchx[kubernetes]"

# install torchx Ray support
pip install "torchx[ray]"
```

### Nightly

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

### Source
Expand All @@ -60,9 +68,15 @@ pip install torchx-nightly
$ 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[kubernetes]
$ pip install -e git+https://github.com/pytorch/torchx.git#egg=torchx[dev]
```

### Docker

TorchX provides a docker container for using as as part of a TorchX role.

See: https://github.com/pytorch/torchx/pkgs/container/torchx

## Contributing

We welcome PRs! See the [CONTRIBUTING](CONTRIBUTING.md) file.
Expand Down