cc_staff
153
edits
PyTorch: Difference between revisions
Jump to navigation
Jump to search
--cpus-per-task is compulsory for srun since Slurm 22.05
m (need to set num threads explicitly in pytorch) |
(--cpus-per-task is compulsory for srun since Slurm 22.05) |
||
| Line 433: | Line 433: | ||
echo "starting training..." | echo "starting training..." | ||
time srun python cifar10-gpu-mps.py --batch_size=512 --num_workers=0 | time srun --cpus-per-task=$SLURM_CPUS_PER_TASK python cifar10-gpu-mps.py --batch_size=512 --num_workers=0 | ||
}} | }} | ||