一、下载win caffe prebuild
直接从GitHub下载prebuild版本免去自行编译的繁琐。
这里没有python 3.6版本的,无奈之下只得选择3.5。
网速比较慢,下载过程中容易断,因此我只得用迅雷下载。
下载完成后,得到caffe.zip
压缩包。解压,文件夹如下:
PS: 如果要安装GPU版本,请确保安装了cuDNN。
二、anaconda环境配置
创建一个新的environment,命名为caffe。
注意这里要选择和刚刚下载的caffe对应的python版本。稍等一会儿,等待环境创建完成。caffe环境的目录一般位于C:\ProgramData\Anaconda3\envs\caffe
。环境创建完成后,会自动安装pip。但是pip的默认镜像源是国外的,下载速度极其缓慢,建议大家更改为豆瓣镜像。具体方法请参考以下文章:windows pip install 使用豆瓣镜像源。
打开anaconda prompt命令行,执行以下命令**caffe环境:
activate caffe
执行以下命令安装依赖库:
pip install cmake ninja numpy scipy protobuf==3.1.0 six scikit-image pyyaml pydotplus graphviz matplotlib
三、使用python接口
- 将下载的压缩包里的
\caffe\python\caffe
文件夹拷贝到C:\ProgramData\Anaconda3\envs\caffe\Lib\site-packages
目录下。 - 打开python环境,用
import caffe
测试:
配置完成。