#!/bin/bash
#SBATCH -t 02:00:00
#SBATCH -p gpu
#SBATCH -N 2
#SBATCH -n 4
#SBATCH --gpus-per-node=gtx1080:2

module load singularity

srun singularity exec --nv Pytorch_DL_HPC.sif python -u DistributedLearning_NCCL.py