Import caffe error

pbu picture pbu · Jan 27, 2015 · Viewed 31.1k times · Source

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

Anoop K. Prabhu picture Anoop K. Prabhu · Jan 30, 2015

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