Fix segfault.

diff --git a/python/caffe/_caffe.cpp b/python/caffe/_caffe.cpp
index 1dbbbc0..257c3d9 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();
 }