[SOLVED] ImportError: No module named caffe

[SOLVED] ImportError: No module named caffe

# Problem

---------------------------------------------------------------------------
ImportError                               Traceback (most recent call last)
<ipython-input-1-da52fc923c25> in <module>()
      6 import matplotlib.pyplot as plt
      7 import sys
----> 8 import caffe
      9 from image_handling import load_dataset
     10 from image_handling import segment_image

ImportError: No module named caffe



# Solution:


import sys
import os

caffe_root = '/root/src/caffe/'  
sys.path.insert(0, caffe_root + 'python')
print(sys.path)