Slurm

class torchx.schedulers.slurm_scheduler.SlurmScheduler(session_name: str)

SlurmScheduler 是 TorchX 的一个调度接口，用于 Slurm。TorchX 假设 Slurm 命令行工具已本地安装，并且作业计费已启用。

每个应用程序定义都通过异构作业使用 sbatch 进行调度。 每个角色的每个副本都会生成一个带有其资源分配和参数的唯一 shell 脚本，然后使用 sbatch 一起启动它们。

日志被写入默认的slurm日志文件。

任何传递给它的调度器选项都会作为SBATCH参数添加到每个副本中。

更多信息请参见：

• https://slurm.schedmd.com/sbatch.html

• https://slurm.schedmd.com/heterogeneous_jobs.html

$ torchx run --scheduler slurm utils.echo --msg hello
slurm://torchx_user/1234
$ torchx status slurm://torchx_user/1234
$ less slurm-1234.out
...

describe(app_id: str) → Optional[torchx.schedulers.api.DescribeAppResponse]

描述指定的应用程序。

Returns

应用程序定义描述或 None 如果应用程序不存在。

schedule(dryrun_info: torchx.specs.api.AppDryRunInfo[torchx.schedulers.slurm_scheduler.SlurmBatchRequest]) → str

与 submit 相同，但接受一个 AppDryRunInfo。 建议实现者实现此方法而不是直接实现 submit，因为 submit 可以通过以下方式简单地实现：

dryrun_info = self.submit_dryrun(app, cfg)
return schedule(dryrun_info)