From a5fc291fe5861417f0648b9cbd061ea0212ce2d5 Mon Sep 17 00:00:00 2001 From: gongweibao Date: Wed, 13 Nov 2019 14:40:05 +0800 Subject: [PATCH] Use 2 cards for hallreduce unit test. (#21085) use 2 cards test=develop --- python/paddle/fluid/tests/unittests/test_dist_base.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/test_dist_base.py b/python/paddle/fluid/tests/unittests/test_dist_base.py index 9a77edd0dc..98b0085b2d 100644 --- a/python/paddle/fluid/tests/unittests/test_dist_base.py +++ b/python/paddle/fluid/tests/unittests/test_dist_base.py @@ -800,7 +800,7 @@ class TestDistBase(unittest.TestCase): if self.__use_cuda: tr_cmd += " --use_cuda" env.update({ - "CUDA_VISIBLE_DEVICES": "{}".format(trainer_id), + "CUDA_VISIBLE_DEVICES": "{}".format(trainer_id % 2), "PADDLE_TRAINERS_NUM": "{}".format(trainer_num), "PADDLE_TRAINER_ID": "{}".format(trainer_id), "PADDLE_TRAINER_ENDPOINTS": self._ps_endpoints, @@ -813,7 +813,7 @@ class TestDistBase(unittest.TestCase): tr_cmd += " --use_dgc" if self._mp_mode: - env = {"FLAGS_selected_gpus": "{}".format(trainer_id)} + env = {"FLAGS_selected_gpus": "{}".format(trainer_id % 2)} if self._nccl_comm_num > 1: tr_cmd += " --nccl_comm_num {}".format(self._nccl_comm_num) -- GitLab