wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-17 12:40:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

made ddp the default if no backend specified with multiple GPUs (#1789)

Browse Source 
* made ddp the default if no backend specified with multiple GPUs

* fix

* spawn

Co-authored-by: Jirka <jirka.borovec@seznam.cz>
This commit is contained in:
William Falcon
2020-05-12 06:54:23 -04:00
committed by GitHub
co-authored by Jirka
parent acab068c74
commit 10b16dbfab
4 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/models/test_gpu.py
+1
View File
@@ -130,6 +130,7 @@ def test_cpu_slurm_save_load(tmpdir):
trainer.fit(model)
@pytest.mark.spawn
@pytest.mark.skipif(torch.cuda.device_count() < 2, reason="test requires multi-GPU machine")
def test_multi_gpu_none_backend(tmpdir):
"""Make sure when using multiple GPUs the user can't use `distributed_backend = None`."""