mirror of
https://github.com/wassname/ray.git
synced 2026-06-28 09:29:05 +08:00
[sgd] Resource limit lift for GPU test (#8238)
This commit is contained in:
committed by
SangBin Cho
parent
97d6509cf8
commit
c2ef9fee74
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user