今天需要使用caffe的python端口,但是发现虚拟环境识别不出之前编译好的caffe了. 最后发现是环境变量的问题.
参考链接: 解决python在import caffe时出现的no module name _caffe问题
pycaffe编译后的配置问题(ModuleNotFoundError: No module named 'caffe')

虚拟环境识别出编译好的caffe, 有三种方式:

  • 把编译好的build路径下python添加到环境变量;
  • 把译好的build路径下python/caffe内容直接复制到虚拟环境的site-packge目录中;(最推荐)
  • 用python的sys添加环境变量;

方法1: 添加虚拟环境

export PATH=/opt/anaconda/envs/bin:$PATH                                                                                      
export PYTHONPATH=~/ZSZT/Geoffrey/caffe/caffe-master/build/python:$PYTHONPATH

首先添加环境变量, ~/ZSZT/Geoffrey/caffe/caffe-master是git clone下的路径, 应该指定下面的python文件夹.
pycaffe编译后的配置问题(ModuleNotFoundError: No module named 'caffe')

可以识别出:
pycaffe编译后的配置问题(ModuleNotFoundError: No module named 'caffe')

方法2: 直接复制

这种方法最方便, 不需要捣鼓麻烦的环境变量
pycaffe编译后的配置问题(ModuleNotFoundError: No module named 'caffe')

方法3: python程序中添加环境变量

添加的路径为git clone完的路径下的python文件夹
pycaffe编译后的配置问题(ModuleNotFoundError: No module named 'caffe')