tao-run-on-slurm

Automate remote TAO GPU job submission on SLURM clusters via SSH.

83|20|Updated Apr 16, 2026
One-click install
npx skills add https://github.com/NVIDIA-TAO/tao-skill-bank --skill tao-run-on-slurm
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: tao-run-on-slurm
Source: https://github.com/NVIDIA-TAO/tao-skill-bank/tree/main/skills/platform/tao-run-on-slurm
Command: npx skills add https://github.com/NVIDIA-TAO/tao-skill-bank --skill tao-run-on-slurm

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill requires nvidia-tao-sdk[slurm], and includes scripts (resource) and references (resource) components.

What problem does it solve?

This Skill simplifies the process of running TAO training, evaluation, or inference jobs on SLURM GPU clusters, providing a streamlined approach to utilizing remote GPU compute resources.

Core Features & Use Cases

  • Remote Execution: Submit and manage TAO jobs on SLURM clusters from anywhere using SSH.
  • Container Support: Utilizes Pyxis/Enroot containers for containerized execution.
  • Lustre Storage: Access and store data on Lustre-backed storage for high-performance data access.
  • Use Case: If you're a developer or engineer who needs to run TAO jobs on an on-premises or DGX SLURM cluster, this skill can automate the process of job submission, execution, and monitoring.

Quick Start

Run your TAO training job on the SLURM cluster using the following command:

/path/to/tao-run-on-slurm/skill command to run training

Frequently Asked Questions about tao-run-on-slurm

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I run TAO training jobs on a remote SLURM GPU cluster?

To run TAO training jobs on a remote SLURM GPU cluster, you can use this Skill to automate SSH job submission and manage execution via the TAO SDK.

Does this workflow support containerized execution with Pyxis and Enroot on SLURM?

Yes, SLURM job submission with this Skill supports containerized execution by utilizing Pyxis and Enroot containers for isolated GPU workflows.

How do I submit multi-node TAO inference and evaluation jobs to SLURM?

You can submit multi-node TAO inference and evaluation jobs to SLURM by automating remote GPU job submission through SSH using the TAO SDK.

Do I need SSH access and a pre-configured SLURM environment to use this?

Yes, you need SSH access and a pre-configured SLURM environment setup to utilize the automated remote GPU job submission and execution features.

Can I use Lustre-backed storage for high-performance data access during remote SLURM execution?

Yes, you can access and store data on Lustre-backed storage to achieve high-performance data access during TAO training and inference on SLURM clusters.

What is the best way to manage TAO training, evaluation, and inference workflows requiring GPU resources?

The best way to manage TAO workflows requiring GPU resources is automating remote SLURM job submission, which handles training, evaluation, and inference execution.