模块“ caffe”没有属性“ set_mode_gpu”

Question

我已经从源代码安装了caffe。我已经使用了Cmake进行安装。我也更新了各自的路径。 我的caffe根目录是：/ home / ashj / caffe 我将PYTHON路径更新为： export PYTHONPATH=<caffe-home>/python:$PYTHONPATH 通过使用 **export PYTHONPATH=/home/ashj/caffe/python:$PYTHONPATH **

我可以加载导入模块caffe。但是我无法访问caffe中的任何方法或任何层，例如set_mode_gpu（），set_mode_cpu（）或层或参数。我收到如下错误：

当我使用

  

进口咖啡

     

caffe.set_mode_gpu（）

我遇到以下错误：

  

Traceback (most recent call last):

     

File "<stdin>", line 1, in <module>

     

AttributeError: module 'caffe' has no attribute 'set_mode_gpu'

PS：我也尝试过使用this link.中提到的caffe .__ caffe.set_mode_gpu（），但它对我不起作用。 我的系统规格：Ubuntu 18.04

TIA

Answer 1

尝试以下步骤，然后设置python PATH： 您可能已经完成了步骤1和3。

make all 
make pycaffe
make distribute
mkdir ~/python 
mv distribute/python/caffe ~/python

在此之后设置您的PYTHONPATH-这应该是一些目录，例如caffe/python/caffe

Answer 2

虽然可能要迟了，但我遇到了同样的问题，并找到了解决方法：

sys.path.insert(0, '/path/to/caffe/python')
import caffe
caffe.set_mode_gpu()

即，在caffe/pathon之前将sys.path的路径添加到您import caffe。
写了post here进行详细分析，希望对您有所帮助。

---

此问题可能是软件包caffe的路径导致的。

对我来说，如果我从Ubuntu终端执行以下操作，一切都会顺利进行：

但是如果我从Pycharm IDE中进行操作，则会发生错误：

请注意，我以两种方式测试了包caffe的路径，并得到了不同的结果：
-在Ubuntu终端中，即运行正常的方式，我得到

'/home/CVAR-B/softwares/caffe/caffe/python/caffe/__init__.pyc'

这是预期结果； -以Pycharm IDE的方式，即发生错误的方式，我得到了

'/usr/local/lib/python2.7/dist-packages/caffe/__init__.pyc'

这不是预期的结果。

鉴于这一发现，我又做了另一件事来处理该错误：

sys.path.insert(0, '/path/to/caffe/python')
import caffe
caffe.set_mode_gpu()

即，将caffe/pathon的路径添加到sys.path之前的import caffe中。

，结果显示这可以解决：

查看caffe.__file__的结果，现在返回预期的路径。