Requesting either of multiple gpu types with sbatch command


Is there any way to request one of multiple gpu types in slurm (or alternatively exclude some gpu types). I have some jobs that require higher memory that the A100’s so I would like to submit my jobs to the public gpu partition such that it uses either of “v100, p100, a40” or excludes “a100”. Is there any way to do this currently?

I currently use --gres=gpu:p100:1 requirement with the sbatch command.

Thank you so much.

In your case, you can add the following lines to your slurs script:
#SBATCH --gres=gpu:2
#SBATCH --constraint=[xeon-6130|xeon-2640v4]
The constraint options can taken from the CPU model column of the compute-node-specifications given by the nodeinfo command and correspond to the v100 and p100 gpu node type, respectively.