diff --git a/python/setup.py.in b/python/setup.py.in
index 8f9f973d933933695b5b251e426933e22cad6366..6d3e6201dc772356396871283f028be76b3180e2 100644
--- a/python/setup.py.in
+++ b/python/setup.py.in
@@ -184,6 +184,8 @@ packages=['paddle',
           'paddle.jit',
           'paddle.jit.dy2static',
           'paddle.inference',
+          'paddle.inference.contrib',
+          'paddle.inference.contrib.utils',
           'paddle.fluid',
           'paddle.fluid.inference',
           'paddle.fluid.dygraph',