python can't import _caffe module

I suspect this is some stupid set-up problem that I make since no one else seems to have trouble running it..

So I was trying to run the pre-trained network by following the instruction, when I tried to run the following command:

python ../python/caffe/detection/detector.py --crop_mode=selective_search --pretrained_model=../examples/imagenet/caffe_reference_imagenet_model --model_def=../examples/imagenet/imagenet_deploy.prototxt _temp/cat.txt _temp/cat.h5

I got an error saying:

Traceback (most recent call last):
  File "detector.py", line 29, in <module>
    import caffe
  File "/python/caffe/__init__.py", line 1, in <module>
    from .pycaffe import Net
  File "/caffe/pycaffe.py", line 6, in <module>
    from ._caffe import CaffeNet
ImportError: No module named _caffe

It seems like python is not recognizing the _caffe.cpp file...Would anyone know how to solve this problem?

Thanks tons!

[shelhamer]

Make sure you have done

make pycaffe

and that you have copied the caffe/python/caffe dir into your PYTHONPATH. The wrapper needs to be compiled–just having the cpp isn't enough. Further, just compiling it doesn't do much good if python can't find it, which is why you need to install it somewhere in your python path.

Hi,

Thanks very much for your reply Evan!

I had trouble finding the makefile for pycaffe, the caffe dir is in my
pythonpath, but I do not have the makefile there..am I missing some files
or steps?

On Thu, Mar 27, 2014 at 1:57 AM, Evan Shelhamer notifications@github.comwrote:

Make sure you have done

make pycaffe

and that you have copied the caffe/python/caffe dir into your PYTHONPATH.
The wrapper needs to be compiled-just having the cpp isn't enough. Further,
just compiling it doesn't do much good if python can't find it, which is
why you need to install it somewhere in your python path.

Reply to this email directly or view it on GitHubhttps://github.com//issues/263#issuecomment-38771502
.

[shelhamer]

Please take another look at the compilation section of the installation guide, above all the parts about make pycaffe and make distribute. You need to build everything first from Caffe root dir, then once you have made pycaffe, which yields a _caffe.so module, you place the distribute/python dir in your python path.

Good luck.

[kloudkl]

Maybe #256 will help.

I have done

make all
make pycaffe
make distribute

my PYTHONPATH looks like this:

/.../caffe_temp/python/caffe:/usr/lib64/python2.6:/.../caffe_temp/distribute/python:/usr/lib64/python2.6/site-packages:/.../caffe_temp/python

I have the _caffe.so module under caffe/python/caffe, but when I run

python pycaffe.py

I still get the error

Traceback (most recent call last):
File "pycaffe.py", line 10, in
from ._caffe import CaffeNet

Thanks!

[longjon]

pycaffe.py is not a script that you can run directly; you should, however, be able to import caffe.

[shelhamer]

Please inspect your PYTHONPATH as it looks wrong. You should give absolute paths to the caffe module's parent directory. For instance, in my configuration I did:

# in caffe root dir
make all
make pycaffe
make distribute
# make dir for custom python modules, install caffe
mkdir ~/python
mv distribute/python/caffe ~/python
# set PYTHONPATH (this should go in your .bashrc or whatever
PYTHONPATH=${HOME}/python:$PYTHONPATH

I created a new directory under home folder and it worked fine, thanks!

Out of curiosity, I wasn't able to find any documentation in terms of how to view the top prediction results for cifar10. Would there be any implementation in caffe that would show the prediction?

[shelhamer]

Closing with resolution of the pycaffe issue. For the network output question, see #281 (comment).

[dkkim930122]

Hello, I had the same (or similar) problem!
I tried to do one of the examples in the Caffe website: Image Classification [notebook].
I was able to import numpy and matplotlib.pyplot, but when I tried to import caffe, I got the following error:

import caffe
Traceback (most recent call last):
File "", line 1, in
ImportError: No module named caffe

I successfully did:
make all
make pycaffe
make distribute

I also added python/caffe & distribute/python/caffe to my PYTHONPATH.

Could someone please help with this issue?
Thanks a lot in advance!

[longjon]

@dkkim930122, I believe you should have caffe/python or caffe/distribute/python in your PYTHONPATH. The caffe folder is itself the module.

[dkkim930122]

@longjon
I appreciate your help!
I think it worked! But now it shows another error message for importing caffe:

import caffe
Traceback (most recent call last):
File "", line 1, in
File "/home/dkkim930122/caffe-master/distribute/python/caffe/init.py", line 1, in
from .pycaffe import Net, SGDSolver
File "/home/dkkim930122/caffe-master/distribute/python/caffe/pycaffe.py", line 10, in
from ._caffe import Net, SGDSolver
ImportError: libmkl_rt.so: cannot open shared object file: No such file or directory

I tried: export LD_LIBRARY_PATH=/opt/intel/mkl/lib/intel64:/usr/local/cuda/lib64
But, this didn't work...
Thank you again!