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 模块也可以自动提示所带有的内容了: