tao-run-on-slurm · AI Skill 用途与安装 · iSkills.cc

在 SLURM GPU 集群上远程运行 TAO 训练和推理任务。

分类:部署运维

来源仓库:nvidia/skills

适合用来做什么

  • 使用 sbatch 在 SLURM 集群上提交我的训练任务。
  • 通过 SSH 运行 srun 命令来启动我的 Pyxis 容器。

不适合的情况

  • Do not use SLURM for local files that exist only on the agent machine; data and outputs must be reachable from the cluster.

安装 tao-run-on-slurm

npx skills add https://www.iskills.cc/skills/nvidia/skills/tao-run-on-slurm/SKILL.md

在本机终端执行安装命令。安装前请阅读原始说明,确认所需工具、账号和权限。

查看安装步骤 · 查看常见问题