mnist_grid_gpu.sh 702 B

1234567891011121314151617181920212223242526272829303132333435
  1. ##!/bin/bash -l
  2. #$ -l tmem=16G
  3. #$ -l h_rt=50:0:0
  4. # Use this flag for a gpu job
  5. #$ -l gpu=true
  6. #$ -S /bin/bash
  7. #$ -j y
  8. #$ -N
  9. # Useful to redirect logs
  10. #$ -o /home/<ucl-id>/hpc-tutorial/jobs/logs
  11. # Defines array of jobs
  12. #$ -t 1:5
  13. # Max number of jobs running at once
  14. #$ -tc 5
  15. hostname
  16. date
  17. SOURCE_DIR='/home/<ucl-id>/hpc-tutorial'
  18. export PYTHONPATH=$PYTHONPATH:$SOURCE_DIR
  19. cd $SOURCE_DIR || exit
  20. source /share/apps/source_files/cuda/cuda-10.1.source
  21. conda activate hpc-example
  22. CONFIG_PATH='jobs/configs/lr-grid.csv'
  23. ROW=$SGE_TASK_ID
  24. LINE=$(sed -n $((ROW + 1))'{p;q}' "$CONFIG_PATH")
  25. ARGS=($(echo "$LINE"))
  26. LR=${ARGS[0]}
  27. python pl_examples/image_classifier_4_lightning_module.py -model.lr $LR
  28. date