diff --git a/paddle/pybind/exception.h b/paddle/pybind/exception.h
index 12c7df93f617d40b5e028d1ae897ce47197c47c6..70beac146046f74e23f747bab130483901a7d443 100644
--- a/paddle/pybind/exception.h
+++ b/paddle/pybind/exception.h
@@ -13,6 +13,7 @@
    limitations under the License. */
 
 #pragma once
+#include <Python.h>
 #include "paddle/platform/enforce.h"
 #include "pybind11/pybind11.h"
 namespace paddle {