[sgd] Resource limit lift for GPU test (#8238)

This commit is contained in:
Richard Liaw
2020-04-30 00:24:48 -07:00
committed by GitHub
parent 254b1ec370
commit 35eac2671e
@@ -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()