#SBATCH --nodes=<some-value>
: Requests a specific number of nodes to run your job on.#SBATCH --mem=<some-value>
: Specifies the amount of RAM required.#SBATCH --time=<some-value>
: Sets the maximum runtime.#SBATCH --output=<some-value>
: Directs the job's output to a specific file.Note: These bullets are just for a better basic understanding on the topic. Complete examples and line-by-line explanations are provided further down in this page.
Queues (or partitions in Slurm terminology) are categories within the cluster that organize jobs based on their resource requirements, priority, and other factors.
Our cluster is partitioned into the following categories:
Choosing the right partition ensures your job is queued in an environment suited to its needs, and can potentially reduce wait times.
In this example we are going to run a Python program with specified resource limits through our batch script.
Create two files, one containing your .sbatch
script and the other containing your .py
program:
I'm going to name them my_script.sbatch
and my_script.py
.
Then add the following to my_script.sbatch
:
#!/bin/bash
#SBATCH --job-name=test_job
#SBATCH --output=test_job.out
#SBATCH --error=test_job.err
#SBATCH --nodes=1
#SBATCH --time=10:00
#SBATCH --mem=1G
module load python3
python3 my_script.py
And add a simple print
statement to my_script.py
:
print("Hello World!")
NOTE: In this example it's assumed that both your batch and Python script files are in the same directory. If that is not the case, please make sure you are providing the full path (starting from the root directory /
) to your Python file, inside of your script. For example:
python3 /path/to/python_script/my_script.py
Now let's walk through my_script.sbatch
line by line to see what each directive does.
#!/bin/bash
: This line needs to be included at the start of all your batch scripts. It basically specifies the script to be run with the bash
shell.Lines 2-7 are your SBATCH
directives. These lines are where you specify different options for your job including its name, output and error files path/name, list of nodes you want to use, resource limits, and more if required. Let's walk through them line by line:
#SBATCH --job-name=test_job
: This directive gives your job a name that you can later use to easier track and manage your job when looking for it in the queue. In this example, we've called it test_job
. You can read about job management at Monitoring jobs.#SBATCH --output=test_job.out
: Used to specify where your output file is generated, and what it's going to be named. In this example, we have not provided a path, but only provided a name. When you use the --output
directive without specifying a full path, just providing a filename, Slurm will store the output file in the current working directory from which the sbatch
command was executed.#SBATCH --error=test_job.err
: Functions similar to --output
except it contains error messages generated during the execution of your job, if any. The .err
file is always going to be generated even if your job execution is successful; however, it's going to be empty if there are no errors.#SBATCH --nodes=1
: Specifies your job to run on one available node. This directive basically tells the scheduler "Run my job on any available node you find, and I don't care which one". It's also possible to specify the name of the node(s) you'd like to use which we will cover in future examples.#SBATCH --time=10:00
: This line specifies how long you want your job to run, after it's out the queue and starts execution. In this case, the job will be terminated after 10 minutes. Acceptable time formats include mm
, mm:ss
, hh:mm:ss
, days-hh
, days-hh:mm
and days-hh:mm:ss
.#SBATCH --mem=1G
Specifies the maximum main memory required per node. In this case we set the cap to 1 gigabyte. If you don't use a memory unit, Slurm automatically uses MegaBytes: #SBATCH --mem=4096
requests 4096MB of RAM. If you want to request all the memory on a node, you can use --mem=0
.After the last #SBATCH
directive, commands are ran like any other regular shell script.
module load python3
: Loads necessary files and modules in order for the command python3
to be valid when used. Please refer to Environment modules for more detail on how the command module
works.python3 my_script.py
: Just like any other python3
command, this line runs the my_script.py
file using Python. Later, the output(s) and/or error(s) of this operation is written to the files we have specified in our directives.This script as discussed previously, is a non-interactive job. Non-interactive jobs are submitted to the queue with the use of the sbatch
command. In this case, we submit our job using sbatch my_script.sbatch
.
You can find more job examples where we run TensorFlow and PyTorch containers at the Apptainer page.
Interactive jobs are those were the user needs to provide input to the application through an interactive pseudo-terminal. For example, this includes the shell and Ncurses-based TUI utilities.
To start an interactive job, you use the srun
command with specific parameters that define your job's resource requirements. Here's an example:
srun --pty --nodelist=cn01 --ntasks=1 --cpus-per-task=1 --time=01:00:00 --mem=4G /bin/bash
Here's a break down of what each part of this command does: **All of these options except the command srun
and the option --pty
are discussed in great detail under the "Batch jobs" section of this page. Refer back to them if you need a more comprehensive explanation.
srun
: The command used to start an interactive session on the cluster.--pty
: Requests a pseudo-terminal, which is necessary for interactive sessions.--nodelist=cn01
: Specifies the node on which to run the interactive session. Here, cn01 is used as an example.--ntasks=1
: Allocates resources for one task.--cpus-per-task=1
: Assigns one CPU to the task. This is enough for tasks that don't require parallel processing across multiple CPUs.--time=01:00:00
: Sets the maximum duration of the interactive session to 1 hour.--mem=4G
: Specifies that the job requires 4 gigabytes of memory./bin/bash
: After allocating the requested resources, srun starts a Bash shell in the interactive session.When your job exits the queue, you will see your shell prompt change, as you receive a shell on the allocated compute node. This shift means that you're now directly interacting with the HPC environment through an interactive session and you are now enabled to execute commands and run applications using the cluster's computational resources in real-time.
srun
Options for Interactive jobsSpecifying a Partition: If your job has specific resource requirements, you may want to specify a partition that matches those needs.
--partition=your_partition_name
Allocating GPUs: For tasks requiring GPU acceleration, you can request specific GPUs.
--gres=gpu:2
This option requests 2 GPUs for your job. Adjust the number according—and within the supported limits—to your task's requirements.
Exclusive Node Access: To ensure that no other jobs share your allocated node, you can request exclusive access.
--exclusive
Memory per CPU: If your job requires a specific amount of memory per CPU, this can be specified.
--mem-per-cpu=4G
This option requests 4 GB of memory per allocated CPU.
Quality of Service (QoS): For prioritizing jobs, you can specify the Quality of Service.
--qos=your_qos
Email Notifications: Stay informed about your job's status with email notifications.
--mail-type=ALL
--mail-user=your_email@example.com
This configuration sends an Email to the specified address at the start, completion, and failure of the job.
To submit an interactive job, simply run the srun
command with your desired options directly in the terminal. For example:
srun --pty --nodelist=cn01 --ntasks=1 --cpus-per-task=1 --time=01:00:00 --mem=4G /bin/bash
Once submitted, you'll be placed in an interactive shell on the allocated node when resources become available. Your prompt will change to indicate you're on the compute node.
To submit an array job, you use the --array
as a part of your sbatch
. This option specifies a range of indices that Slurm uses to create multiple tasks from a single job submission. Each task in the array is assigned a unique SLURM_ARRAY_TASK_ID that can be used within your scripts to differentiate between them.
Suppose you have a dataset split into multiple files and you want to process each file independently. Instead of submitting a separate job for each file, you can submit a single array job where each task processes a different file.
Now, let's have a simple, low-resource Python task that you can run as part of an array job, let's create a Python script that generates a basic report based on the SLURM_ARRAY_TASK_ID
. This script will read from a specific input file based on the task ID, perform a simple operation (like counting the number of lines or words), and output the results to a file.
First, let's create your input files. You can make these as simple text files with a few lines of content. For example: input1.txt
, input2.txt
, and input3.txt
.
Each file could contain a few lines of arbitrary text. You can create these files manually or use the command line:
echo "This is a simple file." > input1.txt
echo "This file contains\nseveral lines of text." > input2.txt
echo "Each file will have\na different\nnumber of lines." > input3.txt
Now here's the content of the Python script, process_data.py
, which reads from one of these input files based on the SLURM_ARRAY_TASK_ID
and counts the number of lines:
import sys
# Get the task ID from the command line arguments
task_id = sys.argv[1]
# Construct the filename based on the task ID
filename = f"input{task_id}.txt"
# Try to open and read the file
try:
with open(filename, 'r') as file:
lines = file.readlines()
num_lines = len(lines)
# Output the results
output_filename = f"output{task_id}.txt"
with open(output_filename, 'w') as outfile:
outfile.write(f"File: {filename}\nNumber of lines: {num_lines}\n")
print(f"Processed {filename} successfully.")
except FileNotFoundError:
print(f"File {filename} not found.")
This script basically takes an argument from the command line (Expected to be the SLURM_ARRAY_TASK_ID
). Then constructs a filename from this ID, reads the corresponding input file, counts its lines, writes the count to an output file, and in case of any missing files it handles them by printing a message instead of crashing.
Finally, to run this script as part of an array job on 3 files, adjust the --array
option in your Slurm script (process_array.sbatch
) to 1-3
.
#!/bin/bash
#SBATCH --job-name=array_job
#SBATCH --output=array_job_%A_%a.out
#SBATCH --error=array_job_%A_%a.err
#SBATCH --array=1-3
#SBATCH --mem=1G
#SBATCH --time=00:10:00
module load python3
python3 process_data.py $SLURM_ARRAY_TASK_ID
In the context of Slurm job submission scripts, %A and %a are special placeholders used within directives like –output and –error to dynamically generate filenames based on the job's array ID and the individual task ID within the array. Here's what each placeholder represents:
%A
: This placeholder is replaced by the Slurm job array's ID. The job array ID is a unique identifier assigned by SLURM to the entire array job at the time of submission. It helps you group and identify all tasks belonging to the same array job.%a
: This placeholder is substituted with the specific task ID within the job array. Since an array job consists of multiple tasks, each with a unique task ID (determined by the --array
option when the job is submitted), %a
allows you to create distinct output or error files for each task, making it easier to troubleshoot and analyze the results of individual tasks.For example, if you submit an array job with the –array=1-10 option and use the following in your script:
#SBATCH --output=job_output_%A_%a.out
#SBATCH --error=job_error_%A_%a.err
Slurm will create separate output and error files for each of the ten tasks in the array. If the array job's ID is 12345, the files for the first task will be named job_output_12345_1.out and job_error_12345_1.err, the files for the second task will be job_output_12345_2.out and job_error_12345_2.err, and so on.
Now submit this job using sbatch process_array.sbatch
and you must see 6 different output files (3 ending in .out
and 3 in .err
). The .out
files each contain the content of the relevant text file they read from, and the .err
files are expected to be empty if everything has run smoothly.
To submit an array job, use the sbatch
command with your Slurm script that includes the --array
option. For example:
sbatch process_array.sbatch
This will submit the entire array job. Slurm will then manage the execution of individual tasks within the array based on available resources.
First, you need to create your MPI program. In this example, we are calling it mpi_hello_world.c
. This program initializes the MPI environment, gets the rank of each process, the total number of processes, and the name of the processor, then prints a greeting from each process.
#include <mpi.h>
#include <stdio.h>
int main(int argc, char** argv) {
MPI_Init(NULL, NULL);
int PID;
MPI_Comm_rank(MPI_COMM_WORLD, &PID);
int number_of_processes;
MPI_Comm_size(MPI_COMM_WORLD, &number_of_processes);
char processor_name[MPI_MAX_PROCESSOR_NAME];
int name_length;
MPI_Get_processor_name(processor_name, &name_length);
printf("Hello MPI user: from process PID %d out of %d processes on machine %s\n", PID, number_of_processes, processor_name);
MPI_Finalize();
return 0;
}
Next, create the compilation and execution script.
Create a Bash script named mpi_hello_world.sh to compile and run the MPI program. This script takes a parameter for the number of processes to spawn.
#!/bin/bash
SRC=mpi_hello_world.c
OBJ=mpi_hello_world
NUM=$1
mpicc -o $OBJ $SRC
mpirun -n $NUM ./$OBJ
This script compiles the MPI program using mpicc
and runs it with mpirun
, and specifies the number of processes with -n
.
Next, prepare a Slurm batch job script named job-test-mpi.sbatch
to submit your MPI job. This script requests cluster resources and runs your MPI program through mpi_hello_world.sh
:
#!/bin/bash
#SBATCH --job-name=mpi_job_test
#SBATCH --output=result.txt
#SBATCH --error=error.txt
#SBATCH --nodelist=gpu1,gpu2,cn01
#SBATCH --time=10:00
#SBATCH --mem-per-cpu=1000
module load openmpi4
echo "run mpi program using parallel processes"
sh mpi_hello_world.sh $1
This script sets up a job with the name mpi_job_test, specifies output and error files, requests resources (All 3 nodes from the cluster), and loads the OpenMPI module. It then runs the mpi_hello_world.sh
script and passes the number of processes as an argument.
Submit your parallel MPI job to Slurm using the sbatch
command sbatch job-test-mpi.sbatch
, specifying the desired number of parallel processes with -n
. For example, to run with 8 parallel processes:
sbatch -n 8 job-test-mpi.sh 8
After the job completes, check the output in result.txt
. You should see greetings from each MPI process. The content might look something like this, shortened with ...
for brevity:
Hello MPI user: from process PID 0 out of 8 processes on machine gpu1
...
Hello MPI user: from process PID 7 out of 8 processes on machine cn01