diff --git a/paddle/framework/CMakeLists.txt b/paddle/framework/CMakeLists.txt
index fcfac5a3e68c5c9c1eac8d643f49aee03fa33dcf..c514e41e13589c89e0fe93d4f511edbdbac7cff7 100644
--- a/paddle/framework/CMakeLists.txt
+++ b/paddle/framework/CMakeLists.txt
@@ -88,5 +88,6 @@ nv_test(data_device_transform_test SRCS data_device_transform_test.cu
 if(NOT WITH_C_API AND WITH_FLUID)
   file(GLOB FRAMEWORK_HEADERS *.h)
   install(FILES ${FRAMEWORK_HEADERS} DESTINATION include/paddle/framework)
+  install(FILES ${CMAKE_CURRENT_BINARY_DIR}/framework.pb.h DESTINATION include/paddle/framework)
   install(FILES details/cow_ptr.h details/op_registry.h DESTINATION include/paddle/framework/details)
 endif()
diff --git a/paddle/string/CMakeLists.txt b/paddle/string/CMakeLists.txt
index 0fa846c4eda674c116d5d8b03da747724097a164..751776dbb5c00972c0b6893fcfb2e710f3f082d7 100644
--- a/paddle/string/CMakeLists.txt
+++ b/paddle/string/CMakeLists.txt
@@ -5,6 +5,6 @@ cc_test(to_string_test SRCS to_string_test.cc)
 
 if(NOT WITH_C_API AND WITH_FLUID)
   file(GLOB STRING_HEADERS *.h)
-  install(FILES ${STRING_HEADERS} DESTINATION include/paddle/memory)
-  install(FILES tinyformat/tinyformat.h DESTINATION include/paddle/memory/tinyformat)
+  install(FILES ${STRING_HEADERS} DESTINATION include/paddle/string)
+  install(FILES tinyformat/tinyformat.h DESTINATION include/paddle/string/tinyformat)
 endif()