windows+anaconda环境下的caffe python安装配置

时间:2024-03-13 09:14:39

一、下载win caffe prebuild

直接从GitHub下载prebuild版本免去自行编译的繁琐。

windows+anaconda环境下的caffe python安装配置

这里没有python 3.6版本的,无奈之下只得选择3.5。
网速比较慢,下载过程中容易断,因此我只得用迅雷下载。
下载完成后,得到caffe.zip压缩包。解压,文件夹如下:
windows+anaconda环境下的caffe python安装配置

PS: 如果要安装GPU版本,请确保安装了cuDNN。

二、anaconda环境配置

  1. 创建一个新的environment,命名为caffe。
    windows+anaconda环境下的caffe python安装配置
    windows+anaconda环境下的caffe python安装配置
    注意这里要选择和刚刚下载的caffe对应的python版本。稍等一会儿,等待环境创建完成。caffe环境的目录一般位于C:\ProgramData\Anaconda3\envs\caffe

  2. 环境创建完成后,会自动安装pip。但是pip的默认镜像源是国外的,下载速度极其缓慢,建议大家更改为豆瓣镜像。具体方法请参考以下文章:windows pip install 使用豆瓣镜像源

  3. 打开anaconda prompt命令行,执行以下命令**caffe环境:
    activate caffe

  4. 执行以下命令安装依赖库:
    pip install cmake ninja numpy scipy protobuf==3.1.0 six scikit-image pyyaml pydotplus graphviz matplotlib

三、使用python接口

  1. 将下载的压缩包里的\caffe\python\caffe文件夹拷贝到 C:\ProgramData\Anaconda3\envs\caffe\Lib\site-packages目录下。
  2. 打开python环境,用import caffe测试:
    windows+anaconda环境下的caffe python安装配置

配置完成。

Reference