TorchX¶
TorchX 是一个用于 PyTorch 应用程序的通用作业启动器。 TorchX 被设计为具有快速迭代时间，适用于训练/研究，并在您准备就绪时支持端到端的生产 ML 管道。
开始使用？ 请遵循 快速入门指南。
In 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!"
文档¶
与以下内容配合使用¶
调度器
管道