mirror of
https://github.com/wassname/ray.git
synced 2026-10-28 13:21:06 +08:00
[sgd] Resource limit lift for GPU test (#8238)
This commit is contained in:
1 parent
97d6509cf8
commit
c2ef9fee74
1 file changed
+1
-1
@@ -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