From 8c2adf625018bb5932f6fe661c5a5c61ab7114db Mon Sep 17 00:00:00 2001 From: William Falcon Date: Sat, 5 Oct 2019 14:28:08 -0400 Subject: [PATCH] cleaned up demos --- examples/multi_node_examples/README.md | 7 +++++-- examples/multi_node_examples/job_submit.sh | 2 +- 2 files changed, 6 insertions(+), 3 deletions(-) diff --git a/examples/multi_node_examples/README.md b/examples/multi_node_examples/README.md index da4086f8..c6c5299e 100644 --- a/examples/multi_node_examples/README.md +++ b/examples/multi_node_examples/README.md @@ -1,7 +1,10 @@ # Multi-node example -Run this module to launch a job which runs on 2 nodes each using 2 GPUs. +To run this demo which launches a single job that trains on 2 nodes (2 gpus per node), do the following: +1. Log into the jumphost node of your SLURM-managed cluster. +2. Create a conda environment with Lightning and a GPU PyTorch version. +3. Submit this script. ```bash -bash job_submit.sh +bash job_submit.sh your_env_name_with_lightning_installed ``` \ No newline at end of file diff --git a/examples/multi_node_examples/job_submit.sh b/examples/multi_node_examples/job_submit.sh index bac61bf4..ee737198 100755 --- a/examples/multi_node_examples/job_submit.sh +++ b/examples/multi_node_examples/job_submit.sh @@ -8,7 +8,7 @@ #SBATCH --time=0-02:00:00 # activate conda env -conda activate my_env +conda activate $1 # ------------------------- # debugging flags (optional)