batch_medium.slurm

#!/bin/bash
#SBATCH --job-name=dg_512_1_4_4_3_100        # <image_size>,<nb_node>,<nb_MPI_task>,<nb_gpu_per_node>,<number_position>,<number_epoch>
#SBATCH --ntasks=4                          # number of tasks
#SBATCH -N 4                                # number of nodes
#SBATCH --gres=gpu:4                        # number of GPU per node
#SBATCH --cpus-per-task=10                  # number of CPU cores

#SBATCH --hint=nomultithread                # use physical core
#SBATCH --time=03:00:00                     # walltime max
#SBATCH --exclusive
#SBATCH -A qbg@gpu

#SBATCH --output=dg.out   # output filename
#SBATCH --error=dg.err    # error filename

set -x
source env_bench 

srun /gpfslocalsup/pub/idrtools/bind_gpu.sh python $DG_TRAIN