Windows下 Pycaffe 的配置与使用

按照我的上一篇博客:Windows 下用 build_win.cmd 直接编译CPU版caffe 配置完 caffe 以后,因为我们在的 build_win.cmd 文件中设置了 电脑上对应的python 版本,并且 BUILD_PYTHON = 1 的话,编译完成以后 pycaffe 也就一起编译好了 。默认 是BUILD_PYTHON = 1,如果之前误修改为 BUILD_PYTHON = 0,将 0 改为 1,然后重新编译 caffe 就可以了。

配置完成后,我新建一个python工程,想要去使用pycaffe,但是当我在pycharm中使用 import caffe 时就会报错:

这里写图片描述

表示还是找不到 caffe 这个模块。

解决方法1:

之所以编译器不能找到 caffe 模块是因为,pycaffe 的路径没有在系统的环境变量里面,当使用时,编译器无法找到它。所以第一个解决方法就是使用时在文件的开始,在系统的环境变量中加入 caffe/python 的路径:

这里写图片描述

这里我的 caffe 是安装在 F:\Software_download 路径下的,所以将 F:\Software_download\caffe-windows\python 添加到 sys.path 就可以了。虽然通过上面的方法程序可以直接运行,但是可以看见,若在 pyharm 中 import caffe 仍然是报错状态的。这样就无法使用编译器很方便的 TAB键 快捷提示功能。而且每一次有新的程序需要使用 pycaffe 的话都要手动添加一次。

解决方法2:

还是以我的电脑路径为例,将caffe/python的路径 F:\Software_download\caffe-windows\python 在操作系统中新建一个系统环境变量 名字叫 PYTHONPATH :

1.找到环境变量

这里写图片描述

2.新建环境变量

这里写图片描述

3.添加以后

这里写图片描述

完成上述添加以后,重启电脑,或者是注销当前用户,重新登陆一下系统,使环境变量生效。再打开我们的 python 文件:

这里写图片描述

并且 caffe 模块也可以自动提示所带有的内容了:

这里写图片描述