Merge branch 'master' into bugfix/5165_enable_pl_optimizer_refactor

Lightning-AI · Dec 20, 2020 · 196d8b4 · 196d8b4
2 parents 151790d + be3e870
commit 196d8b4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/models/test_gpu.py b/tests/models/test_gpu.py
@@ -49,7 +49,7 @@ def test_multi_gpu_none_backend(tmpdir):
     tpipes.run_model_test(trainer_options, model)
 
 
-@pytest.mark.skipif(not torch.cuda.is_available(), reason="test requires GPU machine")
+@pytest.mark.skipif(torch.cuda.device_count() < 2, reason="test requires multi-GPU machine")
 @pytest.mark.parametrize('gpus', [1, [0], [1]])
 def test_single_gpu_model(tmpdir, gpus):
     """Make sure single GPU works (DP mode)."""