torchx

Торчкс

Torchx - это универсальный запуск работы для приложений Pytorch. Torchx предназначен для быстрого времени итерации для обучения/исследований и поддержки трубопроводов E2E ML ML, когда вы будете готовы.

Torchx в настоящее время поддерживает:

• Kubernetes (EKS, GKE, AKS и т. Д.)
• Седло
• AWS партия
• Докер
• Местный
• Рэй (прототип)
• Партия GCP (прототип)

Нужен планировщик, не перечисленный? Дайте нам знать!

QuickStart

Смотрите гид QuickStart.

Документация

• Стабильная документация
• Ночная документация

Требования

TOCHX:

• Python3 (3,8+)
• Пирог
• Необязательно: 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]

Докер

Torchx предоставляет контейнер Docker для использования как часть роли Torchx.

См.: Https://github.com/pytorch/torchx/pkgs/container/torchx

Внося

Мы приветствуем PRS! Смотрите файл внесения.

Лицензия

Torchx имеет лицензию BSD, как найдено в файле лицензии.