TorchX¶
TorchX 是一个用于 PyTorch 应用程序的通用作业启动器。 TorchX 旨在为训练/研究提供快速的迭代时间,并在您准备好时支持端到端的生产机器学习管道。
入门指南? 请遵循 快速入门指南。
在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!"
文档¶
适用于¶
调度器
示例¶
示例