diff --git a/python/ray/util/sgd/tests/test_torch_runner.py b/python/ray/util/sgd/tests/test_torch_runner.py
index 3a1065aad..30c42d437 100644
--- a/python/ray/util/sgd/tests/test_torch_runner.py
+++ b/python/ray/util/sgd/tests/test_torch_runner.py
@@ -179,7 +179,7 @@ class TestTorchRunner(unittest.TestCase):
 class TestLocalDistributedRunner(unittest.TestCase):
     def setUp(self):
         os.environ.pop("CUDA_VISIBLE_DEVICES", None)
-        ray.init(num_gpus=4)
+        ray.init(num_cpus=10, num_gpus=4)
 
     def tearDown(self):
         clear_dummy_actor()