红联Linux门户
Linux帮助

python命令行中import caffe报错的解决方案

发布时间:2016-08-07 09:09:50来源:linux网站作者:白话先生
一、问题
在成功编译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 JAVA_HOME="~/caffe/python" >> ~/.bashrc
B.使环境变量生效
source ~/.bashrc
第二种方法:通过代码来实现
在每个python代码中使用以下代码:
caffe_root = '~/caffe/python '  
import sys  
sys.path.insert(0, caffe_root + 'python')  
import caffe
 
本文永久更新地址:http://www.linuxdiyf.com/linux/23093.html