spawn

Lightning-AI · May 12, 2020 · fa1bb39 · fa1bb39
1 parent 05a4325
commit fa1bb39
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/models/test_gpu.py b/tests/models/test_gpu.py
@@ -130,6 +130,7 @@ def assert_pred_same():
     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`."""