diff --git a/python/paddle/distributed/cloud_utils.py b/python/paddle/distributed/cloud_utils.py
index fcd883d476bb67a56146e5487b4e33d799bf5a80..345b783d60bb79e99c98c4e9d212aa11cbe91dcc 100644
--- a/python/paddle/distributed/cloud_utils.py
+++ b/python/paddle/distributed/cloud_utils.py
@@ -12,8 +12,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from utils import get_cluster, logger
 import os
+import paddle
+from paddle.distributed.utils import get_cluster, logger
 
 
 def get_cloud_cluster(args_node_ips, args_node_ip, args_port, selected_gpus):
diff --git a/python/paddle/distributed/launch.py b/python/paddle/distributed/launch.py
index 1228333f5d35df922fc1fb91258652341d79315c..ecd1cf0ca7bef6586e4833ce80c48eb08a6ad2ee 100644
--- a/python/paddle/distributed/launch.py
+++ b/python/paddle/distributed/launch.py
@@ -154,10 +154,12 @@ def get_cluster_from_args(args, selected_gpus):
         if free_ports is not None:
             free_ports = list(free_ports)
     else:
+        started_port = 6070
+        if args.started_port is not None:
+            started_port = args.started_port
+
         free_ports = [
-            x
-            for x in range(args.started_port, args.started_port + len(
-                selected_gpus))
+            x for x in range(started_port, started_port + len(selected_gpus))
         ]
 
     return get_cluster(node_ips, node_ip, free_ports, selected_gpus)