深度学习Caffe平台开发环境准备——2.python命令行中import caffe报错的解决方案

一、问题

在成功编译caffe的源码之后，可以在python环境中使用caffe。

在Ubuntu环境下，打开python解释程序，输入import caffe时：出现以下错误

>>>import caffe

   Traceback (most recent call last):

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

   ImportError: No module named caffe

二、解决思路

基本思路是把caffe中的python导入到解释器中

三、解决方法

第一种方法：设置环境变量

在终中输入：

export PYTHONPATH=~/caffe/python   #caffe的路径下面的python

 

则该终端起作用，关掉终端后或重新打开一终端，则失效。

放到配置文件中，可以永久有效果，命令操作如下：
A.把环境变量路径放到 ~/.bashrc文件中

sudo echo export PYTHONPATH="~/caffe/python" >> ~/.bashrc

 

B.使环境变量生效

source ~/.bashrc

 

第二种方法：通过代码来实现

在每个python代码中使用以下代码： (这个方法在写python代码时有用)

 

caffe_root = '~/caffe/python '  

import sys  

sys.path.insert(0, caffe_root + 'python')  

import caffe