diff --git a/python/setup.py.in b/python/setup.py.in
index efb29b08620d9620d1b73f2831fa99ec74a58fa8..3288b6152c0f62978184b791c738a9e39919708e 100644
--- a/python/setup.py.in
+++ b/python/setup.py.in
@@ -185,6 +185,10 @@ else:
         shutil.copy(os.path.dirname('${CBLAS_LIBRARIES}') + '/openblas' + ext_name, libs_path)
         package_data['paddle.libs'] += ['openblas' + ext_name]
 
+if '${WITH_PSLIB}' == 'ON':
+    shutil.copy('${PSLIB_LIB}', libs_path)
+    package_data['paddle.libs'] += ['libps' + ext_name]
+
 if '${WITH_MKLDNN}' == 'ON':
     if '${CMAKE_BUILD_TYPE}' == 'Release' and os.name != 'nt':
         # only change rpath in Release mode.