ImportError: No module named _caffe

来源:互联网 发布:网络电视apk软件下载 编辑:程序博客网 时间:2024/06/02 02:25

1. 官方解决方案

官方解决连接
运行下面这句:

export PYTHONPATH=/path/to/caffe/python:$PYTHONPATH

2. 方案1不管用的情况

我用的anaconda2, 用方法1会导致numpy出问题,下面这句报错。

import numpy

解决方法:
在Python脚本里添加如下:

import syssys.path.append("dir/to/caffe/python")

然后 import caffe 应该就ok了

3. 方案2的变种

方案2每次要添加2句话,我们改为1句(好像也没省什么事。。),方法如下:

  • vi ~/.bashrc
    添加一句:

    export CAFFE_ROOT=/DIR/TO/YOUR/CAFFE/python

    然后 source ~/.bashrc

  • 将下面的代码保存到 caffe_header.py

    import sys, ossys.path.append(os.getenv("CAFFE_ROOT") + "/python")
  • caffe_header.py 文件放到你的 PYTHONPATH 中的一个路径中(使用 echo $PYTHONPATH 查看系统里的 PYTHONPATH 有哪些,随便找一个放)。

  • 每次 import caffe 之前,只需要加上import caffe_header 就ok 了,像下面这样

    import caffe_headerimport caffe

Done!

注: 每次编译caffe make -j 之后要重新编译pycaffe make pycaffe,以防万一。

原创粉丝点击