caffe前前后后配了几天总算是弄好了,各种东西间的依赖真的多,也可能是我太笨了|||,总之我的第一篇博客就用来记录caffe环境搭建吧。
操作系统:windows10
显卡:NVIDIA GeForce RTX 2060
anaconda
python3.5
vs2015
CUDA10.0
cudnn7.4.1
cmake3.15.4
Git
主要参考博客:http://www.pianshen.com/article/1113181753/
1.到这里需要安装好上面我提到的所有软件和工具,需要注意的是CUDA的版本要与自己的GPU兼容,CUDA和cudnn版本也要兼容。下面给两个我用到的链接:
·如何查看显卡支持的CUDA版本
https://blog.csdn.net/wangyjfrecky/article/details/55100963
·tensorflow官网,tensorflow、CUDA、cudnn版本依赖关系(最下面有)
https://tensorflow.google.cn/install/source_windows
2.在anaconda下创建一个python3.5的环境
3.安装caffe:
先以git的方式下载caffe,打开cmd命令行,输入git clone https://github.com/BVLC/caffe.git,会自动在
C:\Users\Administrator目录下下载caffe
git clone https://github.com/BVLC/caffe.git
之后输入cd caffe进入caffe文件夹
cd caffe
输入git checkout windows
git checkout windows
此时不要关闭命令行窗口,去C:\Users\Administrator\caffe\scripts 路径下找 build_win.cmd 文件,用文本编辑器打开(我用的notepad++)做如下更改
其中,MSVC_VERSION = 14表示vs2015,CPU_ONLY = 0表示安装GPU版本,PYTHON_VERSION = 3 表示python3.5,WITH_NINJA = 0 表示不适用ninja编译,注意看绿色注释部分来理解,最后设置一下cuDNN路径(图中172行),改好之后保存。
回到刚刚的命令行窗口,输入scripts\build_win.cmd
scripts\build_win.cmd
等待其运行,应该会出错,有一个叫libraries_v140_x64_py35_1.1.0.tar.bz2下不下来,一直0%,
此时可以停下来,去手动下载一个,我是在这里下载的:https://download.csdn.net/download/qq_28829853/10333650
下载好后放在
这个download文件夹是刚才执行scripts\build_win.cmd指令时生成的,如果没有就创建一个。
-----------
接下来我遇到了一个不常见的问题,报错如下图
我就称之为 error MSB6006吧,感谢这篇博客解决了我的问题:
https://blog.csdn.net/qq_37118111/article/details/93385899
该问题是由于caffe版本不支持CUDA10.0了,需要更改源码。具体步骤如下:
找到C:\Users\Administrator\caffe\cmake路径下的Cuda.cmake文件,用文本编辑器打开,做如下修改
之后对 C:\Users\Administrator\.caffe\dependencies\libraries_v140_x64_py35_1.1.0\libraries\include\boost-1_61\boost\config\compiler 路径下的 nvcc.hpp 作如下修改
之后重新编译(删build文件夹,执行指令scripts\build_win.cmd),到这儿应该就OK了,等着它运行完即可,这一步需要花些时间,可以出去冷静冷静,冷静完回来希望显示器上是这般模样:
恭喜,编译完成,快接近尾声了。(期间会有很多黄色的警告信息,无视即可)
去 C:\Users\Administrator\caffe\build 路径下找到Caffe.sln文件,用VS2015将其打开,右键ALL_BUILD进行生成
这一步又要花些时间,可以提前出去庆祝一下了,当你回来应该是这个样子的:
OK~(如果这一步遇到成功37个,失败1个,跳过0个。 这种情况不要慌,重新生成一次)
最后将 C:\Users\Administrator\caffe\python 路径下的 caffe 文件夹粘贴到 需要用的anaconda虚拟环境下(就是我们一开始创建的那个python3.5的环境),我的路径是
H:\sa_Software\Anaconda\envs\python3.5\Lib\site-packages
放到那里,然后再进anaconda prompt去安装一些需要的依赖包,注意在这个命令行窗口默认是base环境,先要切换环境再安装包,输入 activate 环境名 激活环境。
之后用pip安装一些必要的环境(我已经装好了,截图没截这段)
pip install numpy scipy protobuf six scikit-image pyyaml pydotplus graphviz
最后打开python
然后 import caffe,测试一下
大功告成。
博客注册了好多年了,我发的第一篇博客就留给了这个caffe,因为当时真的被搞得头大
欢迎交流指正,如果觉得写得不错请给我点个赞吧~