srun_2.pbs 491 B

123456789101112131415161718192021
  1. #!/bin/bash
  2. #PBS -N skynet_1
  3. #PBS -A jvb-000-ag
  4. #PBS -l nodes=1:gpus=1,walltime=00:01:00
  5. module --force purge
  6. PATH=$PATH:/opt/software/singularity-3.0/bin/
  7. # set the working directory to where the job is launched
  8. cd "${PBS_O_WORKDIR}"
  9. # Run you script inside a singularity image
  10. IMAGE=/rap/jvb-000-aa/singularityimages/pytorch.simg
  11. FOLDERS=$RAP,$HOME,$SCRATCH
  12. SINGULARITY_EXEC="singularity exec --nv --bind $FOLDERS $IMAGE"
  13. # start your python script
  14. $SINGULARITY_EXEC python mnist.py