import numpy before setting dlopen flags (#928)

pytorch · Mar 5, 2017 · ac9245a · ac9245a
1 parent 60736bd
commit ac9245a
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/torch/__init__.py b/torch/__init__.py
@@ -31,6 +31,13 @@
 # automatically filled by the dynamic loader.
 import os as _dl_flags
 
+# if we have numpy, it *must* be imported before the call to setdlopenflags()
+# or there is risk that later c modules will segfault when importing numpy
+try:
+    import numpy as np
+except:
+    pass
+
 # first check if the os package has the required flags
 if not hasattr(_dl_flags, 'RTLD_GLOBAL') or not hasattr(_dl_flags, 'RTLD_NOW'):
     try: