mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-09 11:32:07 +08:00
cleaning up demos (#312)
* cleaning up demos * Update job_submit.sh * Update README.md
This commit is contained in:
@@ -6,5 +6,5 @@ To run this demo which launches a single job that trains on 2 nodes (2 gpus per
|
||||
2. Create a conda environment with Lightning and a GPU PyTorch version.
|
||||
3. Submit this script.
|
||||
```bash
|
||||
sbatch job_submit.sh your_env_name_with_lightning_installed
|
||||
```
|
||||
sbatch job_submit.sh --env=YourEnv
|
||||
```
|
||||
|
||||
@@ -8,7 +8,7 @@
|
||||
#SBATCH --time=0-02:00:00
|
||||
|
||||
# activate conda env
|
||||
conda activate $1
|
||||
source activate $env
|
||||
|
||||
# -------------------------
|
||||
# debugging flags (optional)
|
||||
@@ -24,4 +24,4 @@ conda activate $1
|
||||
# -------------------------
|
||||
|
||||
# run script from above
|
||||
python multi_node_demo.py
|
||||
srun python multi_node_demo.py
|
||||
|
||||
Reference in New Issue
Block a user