mnist_serial_gpu.sh 467 B

123456789101112131415161718192021222324
  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. hostname
  12. date
  13. SOURCE_DIR='/home/<ucl-id>/hpc-tutorial'
  14. export PYTHONPATH=$PYTHONPATH:$SOURCE_DIR
  15. cd $SOURCE_DIR || exit
  16. source /share/apps/source_files/cuda/cuda-10.1.source
  17. conda activate hpc-example
  18. python pl_examples/image_classifier_4_lightning_module.py
  19. date
  20. ```