Fix segfault.

diff --git a/python/caffe/_caffe.cpp b/python/caffe/_caffe.cpp
index 1dbbbc0f12e94341a93bad585edb8bfeb2a06864..257c3d9de05cb2adee09df861ba0f9034d450281 100644 (file)
--- a/python/caffe/_caffe.cpp
+++ b/python/caffe/_caffe.cpp
@@ -321,5 +321,6 @@ BOOST_PYTHON_MODULE(_caffe) {
   boost::python::class_<vector<CaffeLayer> >("LayerVec")
       .def(vector_indexing_suite<vector<CaffeLayer>, true>());
 
+  Py_Initialize();
   import_array();
 }