From 2daa77aa1062c176c96acab10f016407eaff9fff Mon Sep 17 00:00:00 2001 From: sandyhouse Date: Tue, 23 Feb 2021 20:51:18 +0800 Subject: [PATCH] update, test=develop --- python/paddle/fluid/optimizer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle/fluid/optimizer.py b/python/paddle/fluid/optimizer.py index 9612c87d870..42099fbe8c7 100644 --- a/python/paddle/fluid/optimizer.py +++ b/python/paddle/fluid/optimizer.py @@ -4824,7 +4824,7 @@ class PipelineOptimizer(object): place_list = [] for dev in device_list: - dev_index = int(dev.split(":")[1]) + dev_index = int(dev.split(":")[1]) % 8 place_list.append(core.CUDAPlace(dev_index)) # Step6: Split startup program -- GitLab