module'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：我也曾尝试使用caffe .__ caffe.set_mode_gpu（），如this link.中所述，但它对我不起作用。我的系统规格：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路径的结果。

对我来说，如果我从Ubuntu终端执行以下操作，一切都会很好：caffe

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

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

caffe

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

'/home/CVAR-B/softwares/caffe/caffe/python/caffe/__init__.pyc'
这不是预期的结果。 

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

'/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）}

请参阅的结果，现在返回预期的路径。