diff --git a/python/setup.py.server.in b/python/setup.py.server.in
index 0d3cbd34a1d2280375dedcf8b1b24055e7a70895..b602f129253e8f9d55ac17175e387f2232182766 100644
--- a/python/setup.py.server.in
+++ b/python/setup.py.server.in
@@ -29,7 +29,7 @@ util.gen_pipeline_code("paddle_serving_server")
 
 REQUIRED_PACKAGES = [
     'six >= 1.10.0', 'protobuf >= 3.11.0', 'grpcio >= 1.28.1', 'grpcio-tools >= 1.28.1',
-    'paddle_serving_client', 'flask >= 1.1.1', 'paddle_serving_app', 'func_timeout'
+    'paddle_serving_client', 'flask >= 1.1.1', 'paddle_serving_app', 'func_timeout', 'pyyaml'
 ]
 
 packages=['paddle_serving_server',
diff --git a/python/setup.py.server_gpu.in b/python/setup.py.server_gpu.in
index c633e674af182999c3f9d34698f0cc6aec4adf75..7379bcb83a4e48913d22442543e0f65f93d2fa5f 100644
--- a/python/setup.py.server_gpu.in
+++ b/python/setup.py.server_gpu.in
@@ -29,7 +29,7 @@ util.gen_pipeline_code("paddle_serving_server_gpu")
 
 REQUIRED_PACKAGES = [
     'six >= 1.10.0', 'protobuf >= 3.11.0', 'grpcio >= 1.28.1', 'grpcio-tools >= 1.28.1',
-    'paddle_serving_client', 'flask >= 1.1.1', 'paddle_serving_app', 'func_timeout'
+    'paddle_serving_client', 'flask >= 1.1.1', 'paddle_serving_app', 'func_timeout', 'pyyaml'
 ]
 
 packages=['paddle_serving_server_gpu',