Difference between revisions of "Basic Usage: GPU Based Resources With Slurm"

Revision as of 11:30, 8 May 2025

Introduction

Simple commands with SLURM

You can obtain information on the Slurm "Partitions" that accept jobs using the sinfo command

   $ sinfo
   PARTITION AVAIL  TIMELIMIT  NODES  STATE NODELIST
   test         up       1:00      1   idle gnt-usiu-gpu-00.kenet.or.ke
   gpu1         up 1-00:00:00      1   idle gnt-usiu-gpu-00.kenet.or.ke
   normal*      up 1-00:00:00      1   idle gnt-usiu-gpu-00.kenet.or.ke

The test partition is reserved for testing, with a very short time limit. The normal partition is to be used for CPU only jobs, and the gpu1 queue is reserved for GPU jobs. Both production partitions have a time limit of 24 hours at a time for individual jobs.

Showing The Queue

The squeue slurm command will list all submitted jobs, and will give you an indication of how busy the cluster is, as well as the status of all running or waiting jobs. Jobs that are complete will exit the queue and will not be in this list.

   $ squeue 
   JOBID PARTITION     NAME     USER ST       TIME  NODES NODELIST(REASON)
    63    normal     gpu1   jotuya  R       0:03      1 gnt-usiu-gpu-00.kenet.or.ke
   $

Submitting Your first GPU Job

Create a submission script for Quantum Espresso

You require a submission script, which is a plain text file with all the instructions for the command you intend to run. Create a working directory in your scratch directory:

 cd ~/localscratch/
 mkdir test

and in this directory we will place the following text content in a file:

 #!/bin/bash

 #SBATCH -J  gputest               # Job name
 #SBATCH -o job.%j.out         # Name of stdout output file (%j expands to jobId)
 #SBATCH -e %j.err             # Name of std err
 #SBATCH --partition=gpu1    # Queue
 #SBATCH --nodes=1             # Total number of nodes requested
 #SBATCH --gres=gpu:1             # Total number of gpus requested
 #SBATCH --cpus-per-task=1     # 
 #SBATCH --time=00:03:00        # Run time (hh:mm:ss) - 1.5 hours
  
 # Launch MPI-based executable
 module load applications/qespresso/7.3.1 
 
 cd $HOME/localscratch/test
 mpirun -np 1  pw.x <input.in > output.out

Put this in a file called *test.slurm*

Submitting the Job to the Queue

The slurm sbatch command provides the means to submit batch jobs to the queue:

   $ sbatch  test.slurm 
   Submitted batch job 64
   $

This will run the named program on a single GPU, note that the GPU acceleration is built into the program, if the program itself does not support GPU acceleration, attempting to run on the GPU will fail.

Next: Intermediate usage: PyTorch and Tensorflow

Up: HPC_Usage