Remind users to submodule update.

Signed-off-by: Edward Z. Yang <ezyang@fb.com>
pytorch · Aug 30, 2017 · a03e5cb · a03e5cb
1 parent 466f0a8
commit a03e5cb
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -277,6 +277,13 @@ def run(self):
 cwd = os.path.dirname(os.path.abspath(__file__))
 lib_path = os.path.join(cwd, "torch", "lib")
 
+# Check if you remembered to check out submodules
+gloo_cmake = os.path.join(lib_path, "gloo", "CMakeLists.txt")
+if not os.path.exists(gloo_cmake):
+    print("Could not find {}".format(gloo_cmake))
+    print("Did you run 'git submodule update --init'?")
+    sys.exit(1)
+
 tmp_install_path = lib_path + "/tmp_install"
 include_dirs += [
     cwd,