Graphics processing unit (GPU) were originally built to to handle computer graphics processing. There is increasing use of these to perform computational tasks previously handled by the central processing unit (CPU). GPUs operate at lower frequencies than CPUs, but have many more cores. Certain computational tasks can achieve magnitudes of performance improvement with GPU-CPU framework.
The 8 available gpu nodes on the Coeus were out-fitted with a Nvidia A40 and RTXA5000 GPUs.
Gpu nodes
2 X AMD EPYC 7502P 32-Core Processor
128 - 1024 GB ECC RAM
1.7TB local scratch disk
Nvidia A40 or Nvidia RTX A5000 GPU
100Gbps ETHERNET networking
You can start an interactive session as follows:
salloc -N 1 -p gpu --gres=gpu:1 srun --pty bash
You can select specific gpu that you want to use as follows:
--gres=gpu:<gpu_type>:1
Where gpu type is either a40 or rtx
Here's an sbatch example with a trivial cuda program (https://devblogs.nvidia.com/easy-introduction-cuda-c-and-c/)
#!/bin/bash
#SBATCH --job-name saxpy
#SBATCH --nodes 1
#SBATCH --partition himem
#SBATCH --output saxpy_%j.txt
#SBATCH --error saxpy_%j.err
#SBATCH --gres=gpu:1
srun /home/username/examples/cuda/saxpy