Import caffe error

Question

i compiled caffe successfully in my ubuntu machine but cannot import in python.

Caffe is installed /home/pbu/Desktop/caffe

i tried adding the /home/pbu/caffe/python path to sys.path.append, still not working

i am trying to import caffe

root@pbu-OptiPlex-740-Enhanced:/home/pbu/Desktop# python ./caffe/output.py
Traceback (most recent call last):
  File "./caffe/output.py", line 13, in <module>
    import caffe
  File "/home/pbu/Desktop/caffe/python/caffe/__init__.py", line 1, in <module>
    from .pycaffe import Net, SGDSolver
  File "/home/pbu/Desktop/caffe/python/caffe/pycaffe.py", line 10, in <module>
    from ._caffe import Net, SGDSolver
ImportError: No module named _caffe

Answer 1

This happens when you have not run make for the python files separately.

Run make pycaffe soon after running make in the Caffe directory.

You may have to set the path to the python library correctly in Makefile.config

Answer 2

Adding to the above best answer. After you run make for python files by running make pycaffe where you ran your previous makes. Then you have to export that python path by running export PYTHONPATH=<path-to-caffe>/python. You can choose to run this everytime before running a python code which utilizes caffe or add it to your ~/.bashrc.