By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement . We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

root@63b86f5db157:/workspace/gitzkl/RRPN# python ./tools/rotation_demo.py
Traceback (most recent call last):
File "./tools/rotation_demo.py", line 18, in
from fast_rcnn.test import im_detect
File "/workspace/gitzkl/RRPN/tools/../lib/fast_rcnn/test.py", line 16, in
import caffe
File "/workspace/gitzkl/RRPN/tools/../caffe-fast-rcnn/python/caffe/ init .py", line 1, in
from .pycaffe import Net, SGDSolver, NesterovSolver, AdaGradSolver, RMSPropSolver, AdaDeltaSolver, AdamSolver
File "/workspace/gitzkl/RRPN/tools/../caffe-fast-rcnn/python/caffe/pycaffe.py", line 13, in
from ._caffe import Net, SGDSolver, NesterovSolver, AdaGradSolver,
ImportError: No module named _caffe

但是在该容器下 跑 python
import caffe
正常运行,没有报错,

/opt/caffe/build/tools/caffe
cd /opt/caffe/build
make pycaffe
[100%] Built target pycaffe

python ./tools/rotation_demo.py,出现上述错误,

环境是docker 建立的 nvidia-docker run -ti bvlc/caffe:gpu caffe --version