SLURM GPU

Revision as of 17:11, 21 June 2016 by Botto (talk | contribs)


GPU Jobs

Research Computing hosts 28 compute nodes with GPU capabilities. Requesting GPU resources requires additional parameters to be added to your job script in order to land on the right equipment and ensure that your job is dispatched correctly.


Basic GPU Job

For basic GPU jobs, where you will be using a single CPU thread and a single GPU, the following will be sufficient:


#SBATCH --name=gpu_job
#SBATCH --time=01:00:00
#SBATCH --gres=gpu:1
./GpuEnabledExe

You can request up to two GPUs be used by a single node job. This can be requested by calling


#SBATCH --gres=gpu:2


MPI GPU Jobs

On CIRCE, MPI parallel jobs with GPU support require extra care when crafting your request. You MUST use the nodes,ntasks-per-node parameters for your job request. In this case, every node you select (up to 20) will get you an additional GPU. So, lets say I want to use 8 GPUs with MPI, each with a single CPU thread:

#SBATCH --name=myMPIGPUjob
#SBATCH --time=01:00:00
#SBATCH --nodes=8
#SBATCH --ntasks-per-node=1
#SBATCH --gres=gpu:1
./MpiGpuExe

The largest possible GPU job is expressed below (this may take quite some time to dispatch)

#SBATCH --name=myMPIGPUjob
#SBATCH --time=01:00:00
#SBATCH --nodes=20
#SBATCH --ntasks-per-node=2
#SBATCH --gres=gpu:2
./MpiGpuExe

This will request the use of 40 GPUs for an MPI application.


Specify a GPU model

Since there are 3 generations of GPUs available, you may want to specify which you'd like to use. The versions and flags required are listed below:

Model Nickname Constraint Number Available
T10 Tesla gpu_T10 2
M2070 Fermi gpu_M2070 8
K20 Kepler gpu_K20 42