From fa1bb390927b01ab666dd3b8f7fe6f845568abff Mon Sep 17 00:00:00 2001 From: Jirka Date: Tue, 12 May 2020 09:42:12 +0200 Subject: [PATCH] spawn --- tests/models/test_gpu.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/models/test_gpu.py b/tests/models/test_gpu.py index 6eafc19d863ee2..f75b0a1f1a582c 100644 --- 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`."""