mirror of
https://github.com/wassname/ray.git
synced 2026-06-28 01:16:06 +08:00
[sgd] Resource limit lift for GPU test (#8238)
This commit is contained in:
@@ -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