ImportError: No module named

xiaoxiao2021-02-28  96

1. 官方解决方案

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

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

2. 方案1不管用的情况

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

import numpy

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

import sys sys.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, os sys.path.append(os.getenv("CAFFE_ROOT") + "/python")

将caffe_header.py 文件放到你的 PYTHONPATH 中的一个路径中(使用 echo $PYTHONPATH 查看系统里的 PYTHONPATH 有哪些,随便找一个放)。

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

import caffe_header import caffe

Done!

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

转载请注明原文地址: https://www.6miu.com/read-55754.html

最新回复(0)