torchx

Torchx

Torchx adalah peluncur kerja universal untuk aplikasi Pytorch. Torchx dirancang untuk memiliki waktu iterasi cepat untuk pelatihan/penelitian dan dukungan untuk pipa E2E ML pipa saat Anda siap.

Torchx saat ini mendukung:

• Kubernetes (EKS, GKE, AK, dll)
• Slurm
• Batch AWS
• Buruh pelabuhan
• Lokal
• Ray (prototipe)
• Batch GCP (prototipe)

Butuh penjadwal yang tidak terdaftar? Beri tahu kami!

QuickStart

Lihat panduan QuickStart.

Dokumentasi

• Dokumentasi yang stabil
• Dokumentasi malam hari

Persyaratan

Torchx:

• Python3 (3.8+)
• Pytorch
• Opsional: Docker (Diperlukan untuk Penjadwal Berbasis Docker)

Penjadwal tertentu mungkin memerlukan persyaratan khusus penjadwal. Lihat Instalasi untuk info.

Instalasi

Stabil

 # 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] "

Malam

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

Sumber

 # 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]

Buruh pelabuhan

Torchx menyediakan wadah Docker untuk digunakan sebagai bagian dari peran Torchx.

Lihat: https://github.com/pytorch/torchx/pkgs/container/Torchx

Berkontribusi

Kami menyambut PRS! Lihat file yang berkontribusi.

Lisensi

Torchx berlisensi BSD, seperti yang ditemukan dalam file lisensi.