Shortcuts

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!"

文档

适用于

示例

组件库

运行时库

应用程序(运行时)

参考