Caffe未在python中导入

时间:2019-03-11 22:52:10

标签: python import module installation caffe

Ubuntu 18.04

Python 2.7

我的问题是即使安装了caffe模块,我也无法导入它。我相信这是路径/环境变量的问题。

--requirements_file reqs.txt

here所述,我运行了rivaldo4t@Rivaldo-OS3:~$ python Python 2.7.15rc1 (default, Nov 12 2018, 14:31:15) [GCC 7.3.0] on linux2 Type "help", "copyright", "credits" or "license" for more information. >>> import caffe Traceback (most recent call last): File "<stdin>", line 1, in <module> ImportError: No module named caffe 命令来安装预编译的Caffe,它成功完成了。当我在终端中输入caffe时,我会得到

sudo apt install caffe-cpu

这意味着caffe已正确安装。如here所述,我尝试将rivaldo4t@Rivaldo-OS3:~$ caffe caffe: command line brew usage: caffe <command> <args> commands: train train or finetune a model test score a model device_query show GPU diagnostic information time benchmark model execution time 变量设置为Caffe安装目录的位置。所以我做到了,

PYTHONPATH

当我尝试运行echo $ PYTHONPATH时,它显示为空(我仍然不知道为什么)。然后,我尝试使用

更新路径
rivaldo4t@Rivaldo-OS3:~$ which caffe
/usr/bin/caffe

所以现在我在打印sys.path时得到了-

caffe_root = '/usr/bin/caffe/python'
import sys
sys.path.insert(0, caffe_root)

但是我仍然无法在python中导入caffe。我注意到的另一件事是/ usr / bin / caffe不是目录,如果不在这里,我不明白caffe的安装位置。

感谢您的帮助。谢谢

1 个答案:

答案 0 :(得分:0)

当您使用sudo apt install caffe-cpu在ubuntu上安装caffe时,它将仅编译位于_caffe.cpython-36m-x86_64-linux-gnu.so的python 3(/usr/lib/python3/dist-packages/caffe/)的绑定。因此,简短的答案是改用python 3。

长答案是用源代码中的python 2绑定编译caffe。