diff --git a/cmake/generic.cmake b/cmake/generic.cmake
index 24a07c0a24143689e711e5b81df8a221ca95979d..8736d30059e53f6798bf64b5db954630a4760242 100644
--- a/cmake/generic.cmake
+++ b/cmake/generic.cmake
@@ -357,26 +357,3 @@ function(pb_cc_library TARGET_NAME)
   include_directories(${CMAKE_CURRENT_BINARY_DIR})
   cc_library(${TARGET_NAME} SRCS ${proto_srcs})
 endfunction()
-
-function(pb_py_library TARGET_NAME)
-  set(oneValueArgs TARGET_DIR)
-  set(multiValueArgs SRCS)
-  cmake_parse_arguments(pb_py_library "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})
-  if (NOT pb_py_library_TARGET_DIR)
-    set(pb_py_library_TARGET_DIR ${CMAKE_CURRENT_BINARY_DIR})
-  endif()
-  set(py_srcs)
-  foreach(FIL ${pb_py_library_SRCS})
-    get_filename_component(ABS_FIL ${FIL} ABSOLUTE)
-    get_filename_component(FIL_WE ${FIL} NAME_WE)
-    set(cur_py_src ${pb_py_library_TARGET_DIR}/${FIL_WE}_pb2.py)
-    list(APPEND py_srcs "${cur_py_src}")
-    add_custom_command(OUTPUT ${cur_py_src}
-            COMMAND ${PROTOBUF_PROTOC_EXECUTABLE}
-            ARGS "--python_out=${pb_py_library_TARGET_DIR}" "-I" ${CMAKE_CURRENT_SOURCE_DIR} ${ABS_FIL}
-            DEPENDS ${ABS_FIL} protoc
-            COMMENT "Running Python protocol buffer compiler on ${FIL}")
-  endforeach()
-
-  add_custom_target(${TARGET_NAME} ALL DEPENDS ${py_srcs})
-endfunction()
\ No newline at end of file
diff --git a/proto/CMakeLists.txt b/proto/CMakeLists.txt
index 4402f2c89947a1424654854972aa6560fc6ede08..1cf39d69443cfd8006385d9d8f90ae029cabdbd4 100644
--- a/proto/CMakeLists.txt
+++ b/proto/CMakeLists.txt
@@ -1,3 +1,20 @@
 file(GLOB proto_filenames . *.proto)
 pb_cc_library(paddle_proto SRCS ${proto_filenames})
-pb_py_library(gen_proto_py SRCS ${proto_filenames} TARGET_DIR ${PROJ_ROOT}/python/paddle/proto)
+
+set(PROTO_GEN)
+set(PROTO_GEN_PY)
+
+foreach(filename ${proto_filenames})
+    get_filename_component(base_filename ${filename} NAME_WE)
+    set(CUR_PROTO_GEN_PY
+            ${PROJ_ROOT}/paddle/python/paddle/proto/${base_filename}_pb2.py)
+    set(PROTO_GEN_PY
+            ${CUR_PROTO_GEN_PY}
+            ${PROTO_GEN_PY})
+    add_custom_command(OUTPUT ${CUR_PROTO_GEN_PY}
+            COMMAND env ${py_env} ${PROTOBUF_PROTOC_EXECUTABLE} --python_out ${PROJ_ROOT}/python/paddle/proto
+            --proto_path ${PROJ_ROOT}/proto ${PROJ_ROOT}/proto/${filename}
+            DEPENDS ${filename} ${external_project_dependencies})
+endforeach()
+
+add_custom_target(gen_proto_py ALL DEPENDS ${PROTO_GEN_PY})