TorchX¶
TorchX 是 PyTorch 应用程序的通用作业启动器。 TorchX 旨在为培训/研究和支持提供快速迭代时间 用于 E2E 生产 ML 管道。
开始?请遵循快速入门指南。
在 1-2-3 中¶
步骤 1.安装
pip install torchx[dev]
步骤 2。在本地运行
torchx run --scheduler local_cwd utils.python --script my_app.py "Hello, localhost!"
步骤 3。远程运行
torchx run --scheduler kubernetes utils.python --script my_app.py "Hello, Kubernetes!"
文档¶
Works With¶
调度程序
管道