diff --git a/imperative/python/megengine/distributed/server.py b/imperative/python/megengine/distributed/server.py index 581a85c8187eb4866446db9ca5fc95efdb9be5a3..e7875851bc77ca38201ca7119cf9f335c8ca4543 100644 --- a/imperative/python/megengine/distributed/server.py +++ b/imperative/python/megengine/distributed/server.py @@ -6,7 +6,6 @@ # Unless required by applicable law or agreed to in writing, # software distributed under the License is distributed on an # "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -import multiprocessing as mp import threading import time from collections import defaultdict @@ -142,7 +141,7 @@ class Server: def __init__(self, port): self.py_server_port = get_free_ports(1)[0] if port == 0 else port self.mm_server_port = create_mm_server("0.0.0.0", 0) - self.proc = mp.Process( + self.proc = threading.Thread( target=start_server, args=(self.py_server_port, self.mm_server_port), daemon=True,