TorchX¶
TorchX 是 PyTorch 应用程序的通用作业启动器。TorchX 旨在为训练/研究提供快速迭代时间,并在您准备就绪时支持端到端生产 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!"