Submitting GPU jobs
The HPC supports GPU jobs. The HPC includes an increasing number of compute nodes equipped with GPU hardware.
GPU resources on the HPC are available in certain access Slurm accounts (partitions):
GPU Model | Slurm accounts |
---|---|
NVIDIA GeForce GTX 1080 Ti | Owner accounts only |
NVIDIA RTX A4000 | genacc_q , backfill , backfill2 , quicktest |
NVIDIA H100 Tensor Core GPU | Limited access for DSC pilot program |
In addition, if your department, lab, or group has purchased GPU resources, they will be available on your owner-based Slurm account.
Submitting GPU Jobs#
If you wish to submit a job to node(s) that have GPUs, simply add the following line to your submit script:
Nodes contain two to four GPU cards. Specify the number of GPU cards per node you wish to use after the --gpus
directive.
For example, if your job requires four GPU cards, simply specify 4
:
Full Example Submit Script#
The following HPC job will run on a GPU node and simply print information about the available GPU cards:
Your job output should look something like this:
For more information and examples, refer to our CUDA software documentation