Skip to content

Instantly share code, notes, and snippets.

@aribornstein

aribornstein/slurm.config Secret

Created June 21, 2021 08:48
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save aribornstein/8969788f4dc0d4e70809cb0e05e4fb2d to your computer and use it in GitHub Desktop.
Save aribornstein/8969788f4dc0d4e70809cb0e05e4fb2d to your computer and use it in GitHub Desktop.
Download ZIP
Raw
slurm.config
# SLURM SUBMIT SCRIPT
#SBATCH --nodes=2
#SBATCH --gres=gpu:8
#SBATCH --ntasks-per-node=1
#SBATCH --time=02:00:00
# load your Python environment, for example with conda
conda activate your_env_name
# run script
srun python train.py --trainer.num_nodes 2 --trainer.gpus 8
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment