diff --git a/python/paddle/fluid/tests/book_memory_optimization/test_memopt_fit_a_line.py b/python/paddle/fluid/tests/book_memory_optimization/test_memopt_fit_a_line.py
index a7dd3d91d21bb1171627d0a7f8ba85e8d4dac293..7648bb9fe1c8d2dae3590fa67f781743ba297c32 100644
--- a/python/paddle/fluid/tests/book_memory_optimization/test_memopt_fit_a_line.py
+++ b/python/paddle/fluid/tests/book_memory_optimization/test_memopt_fit_a_line.py
@@ -31,10 +31,10 @@ use_nccl = False
 place = fluid.CPUPlace()
 if fluid.core.is_compiled_with_cuda():
     device_type = 'CUDA'
-    use_nccl = True
+    use_nccl = False
     place = fluid.CUDAPlace(0)
 
-places = fluid.layers.get_places(device_count=2, device_type=device_type)
+places = fluid.layers.get_places(device_count=0, device_type=device_type)
 pd = fluid.layers.ParallelDo(places, use_nccl=use_nccl)
 with pd.do():
     x_ = pd.read_input(x)