Submitting GPU Jobs
The Oscar GPUs are in a separate partition to the regular compute nodes. The partition is called gpu
. To see how many jobs are running and pending in the gpu partition, use
Interactive use
To start an session on a GPU node, use the interact
command and specify the gpu
partition. You also need to specify the requested number of GPUs using the -g
option:
Batch jobs
Here is an example batch script for a cuda job that uses 1 gpu and 1 cpu for 5 minutes
To submit this script:
DGX GPU Nodes in the GPU-HE Partition
All the nodes in the gpu-he partition have V100 GPUs. However, two of them are DGX nodes (gpu1404/1405) which have 8 GPUs. When a gpu-he job requests for more than 4 GPUs, the job will automatically be allocated to the DGX nodes.
The other non-DGX nodes actually have a better NVLink interconnect topology as all of them have direct links to the other. So the non-DGX nodes are better for a gpu-he job if the job does not require more than 4 GPUs.
Last updated