cleaned up demos

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
 ```

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)