在成功编译caffe的源码之后,可以在python环境中使用caffe。e
在Ubuntu环境下,打开python解释程序,输入import caffe时:出现以下错误
>>>import caffe
Traceback (most recent call last): File "<stdin>", line 1, in <module>
ImportError: No module named caffe
二、解决思路
基本思路是把caffe中的python导入到解释器中
三、解决方法
放到配置文件中,可以永久有效果,命令操作如下:
A.把环境变量路径放到 ~/.bashrc文件中,打开文件
sudo vim ~/.bashrc
在文件下方写入
export PYTHONPATH=~/caffe/python:$PYTHONPATH
上述语句中 “~” 号表示caffe 所在的根目录。
B.关闭文件,在终端写入下面语句,使环境变量生效
source ~/.bashrc
还有另一种方法,直接在运行的程序前加上如下代码:
import sys
sys.path.append("/home/用户名/caffe/python")
import os,sys,caffe
网友评论