#!/bin/bash
#SBATCH --nodes=1
#SBATCH --ntasks-per-node=2
#SBATCH --time=00:01:00
#SBATCH --gres=gpu
#SBATCH --partition=develbooster
#SBATCH --account=atmlaml

set -e

######## How to run with container #########
module purge

source config.sh

export SRUN_CPUS_PER_TASK="$SLURM_CPUS_PER_TASK"

echo "Running with image $SINGULARITY_IMAGE"
echo "Check that your virtual environment is in the PYTHONPATH:"
srun singularity run $SINGULARITY_IMAGE bash -c ' 
    source activate.sh;
    echo hello world;
    python -c \"import sys; print(sys.path)\"
'
