diff --git a/pdseg/solver.py b/pdseg/solver.py
index 8eea7400b5b3ad56777c6d00577a2725737f2089..1767374720eb1bb0777c6df2fb000254a25b3e14 100644
--- a/pdseg/solver.py
+++ b/pdseg/solver.py
@@ -18,7 +18,10 @@ import paddle.fluid as fluid
 import numpy as np
 import importlib
 from utils.config import cfg
-from paddle.fluid.contrib.mixed_precision.decorator import OptimizerWithMixedPrecison, decorate, AutoMixedPrecisionLists
+try:
+    from paddle.fluid.contrib.mixed_precision.decorator import OptimizerWithMixedPrecison, decorate, AutoMixedPrecisionLists
+except:
+    from paddle.fluid.contrib.mixed_precision.decorator import OptimizerWithMixedPrecision, decorate, AutoMixedPrecisionLists
 
 
 class Solver(object):