diff --git a/python/paddle/fluid/tests/unittests/ir_memory_optimize_net_base.py b/python/paddle/fluid/tests/unittests/ir_memory_optimize_net_base.py
index 84aa6b035272eb004b54c342d869ec2d97165071..bf6adce8acad5dd1342afcf8b64c63f5129540f6 100644
--- a/python/paddle/fluid/tests/unittests/ir_memory_optimize_net_base.py
+++ b/python/paddle/fluid/tests/unittests/ir_memory_optimize_net_base.py
@@ -121,7 +121,7 @@ class TestIrMemOptBase(BuildIrMemOptBase):
             return
 
         baseline_first_loss, baseline_last_loss = None, None
-        for use_cuda in [True, False]:
+        for use_cuda in [True]:
             for use_python_mem_opt in [True, False]:
                 print(
                     'network: {}, use_cuda: {}, use_python_mem_opt: {}, use_ir_mem_opt : {}'.