batch-script.sh

#!/bin/bash

# Sample Slurm job script for Galvani 

#SBATCH -J differometor            # Job name
#SBATCH --ntasks-per-node=1                 # Number of tasks
#SBATCH --cpus-per-task=8          # Number of CPU cores per task
#SBATCH --nodes=1                  # Ensure that all cores are on the same machine with nodes=1
#SBATCH --partition=a100-galvani   # Which partition will run your job
#SBATCH --time=0-02:30
#SBATCH --gres=gpu:1               # (optional) Requesting type and number of GPUs
#SBATCH --mem=50G                  # Total memory pool for all cores (see also --mem-per-cpu); exceeding this number will cause your job to fail.
#SBATCH --output=./jobfiles_out/myjob-%j.out       # File to which STDOUT will be written - make sure this is not on $HOME
#SBATCH --error=./jobfiles_err/myjob-%j.err        # File to which STDERR will be written - make sure this is not on $HOME
#SBATCH --mail-type=END,FAIL            # Type of email notification- BEGIN,END,FAIL,ALL
#SBATCH --mail-user=laurin.sefa@student.uni-tuebingen.de   # Email to which notifications will be sent

# Diagnostic and Analysis Phase - please leave these in.
scontrol show job $SLURM_JOB_ID
pwd
# nvidia-smi # only if you requested gpus
# ls $WORK # not necessary just here to illustrate that $WORK is available here


# Setup Phase
# add possibly other setup code here, e.g.
# - copy singularity images or datasets to local on-compute-node storage like /scratch_local
# - loads virtual envs, like with anaconda
# - set environment variables
# - determine commandline arguments for `srun` calls
source $WORK/Differometor/.venv/bin/activate

# Compute Phase
srun python -m optimization.examples.voyager_benchmark

# Profiling:
# srun python -m cProfile -o profile_output.prof -m optimization.examples.voyager_bo

# For interactive session:
# srun -p 2080-galvani --gres=gpu:1 --pty bash -I