cc_staff
153
edits
Using GPUs with Slurm: Difference between revisions
m
Consistency with OMP_NUM_THREADS
(--cpus-per-task is compulsory for srun since Slurm 22.05, and doesn't hurt for older versions) |
m (Consistency with OMP_NUM_THREADS) |
||
| Line 138: | Line 138: | ||
#SBATCH --time=0-03:00 # time (DD-HH:MM) | #SBATCH --time=0-03:00 # time (DD-HH:MM) | ||
export OMP_NUM_THREADS=$SLURM_CPUS_PER_TASK | export OMP_NUM_THREADS=$SLURM_CPUS_PER_TASK | ||
srun --cpus-per-task= | srun --cpus-per-task=$SLURM_CPUS_PER_TASK ./program | ||
}} | }} | ||