diff --git a/setup.py b/setup.py
index fc058f4288fd0..732cd5b878083 100644
--- a/setup.py
+++ b/setup.py
@@ -849,6 +849,7 @@ def make_relative_rpath(path):
         entry_points=entry_points,
         package_data={
             'torch': [
+                '__init__.pyi',
                 'lib/*.so*',
                 'lib/*.dylib*',
                 'lib/*.dll',