From 35eac2671eeeb65e57f4f2d17e468d779d2fc2b0 Mon Sep 17 00:00:00 2001 From: Richard Liaw Date: Thu, 30 Apr 2020 00:24:48 -0700 Subject: [PATCH] [sgd] Resource limit lift for GPU test (#8238) --- python/ray/util/sgd/tests/test_torch_runner.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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()