[slurm-users] Job requesting two different GPUs on two

[Gestió Servidors]

Hello,

No, with "#SBATCH --gres=gpu:2" SLURM searchs a node with 2 GPUs but I need to run my job in 2 nodes using 2 GPUs but one GPU in each node. If both GPUs are the same, job runs OK, but I want to test run my job in two nodes: one offers a GeForceRTX3080 and the second offers a GeForceRTX2070. In this case, I have been not able to define a "#SBATCH --gres=" correct line.

Thanks.

>>>> Hi,
>>>>
>>>>
>>>>
>>>> Today, doing some tests, I have not got a solution to write a submit script
>>>that requests 2 different GPUs on 2 different nodes. With this simple script:
>>>>
>>>>
>>>>
>>>> #!/bin/bash
>>>> #
>>>> #SBATCH --job-name=N2n4
>>>> #SBATCH --output=N2n4-CUDA.txt
>>>> #SBATCH --gres=gpu:GeForceRTX3080:1
>>>>
>>>> #SBATCH -N 2 # number of nodes
>>>> #SBATCH -n 4 # number of cores
>>>> #SBATCH --partition=cuda.q
>>>>
>>>> module load cuda/11.2
>>>>
>>>> sleep 100
>>>> mpirun /home/caos/druiz/samples-SLURM/OpenMPI/mpihello-3.0.0
>>>>
>>>>
>>>>
>>>> I get a parallel MPI job in two nodes, two process in each node and one
>>>GeForceRTX3080 in each node. However, if I want to request 2 different GPUs,
>>>I can?t write ?#SBATCH --gres=gpu:GeForceRTX3080:1, --
>>>gres=gpu:GeForceRTX2070:1?
>>>> because line ?#SBATCH --gres=? is for each node and, then, a line containing
>>>two ?gres? would request a node with 2 different GPUs. So? is it possible to
>>>request 2 different GPUs in 2 different nodes?
>>>
>>>Have you tried
>>>
>>>  #SBATCH --gres=gpu:2
>>>
>>>?
>>>
>>>Cheers,
>>>
>>>Loris
>>>
>>>--
>>>Dr. Loris Bennett (Hr./Mr.)
>>>ZEDAT, Freie Universit?t Berlin         Email loris.bennett at fu-berlin.de
>>>
>>>
>>>
>>>End of slurm-users Digest, Vol 44, Issue 16
>>>*******************************************