Using GPUs with Slurm: Difference between revisions

Jump to navigation Jump to search
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=6 ./program
srun --cpus-per-task=$SLURM_CPUS_PER_TASK ./program
}}
}}


cc_staff
153

edits

Navigation menu