import numpy before setting dlopen flags (#928)

albanD · soumith · commit ac9245aeb32a · 2017-03-05T14:30:13.000-05:00
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:
