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)