超详细win10+caffe+gpu (NVIDIA GeForce RTX 2060)+vs2015+CUDA10.0+cudnn7.4.1+anaconda+python3.5安装

 一、写在前面

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

下载好后放在

C:\Users\Administrator\.caffe\dependencies\download 路径下(注意是.caffe文件夹)

这个download文件夹是刚才执行scripts\build_win.cmd指令时生成的,如果没有就创建一个。

重点:每次出错,需要删除scripts目录下的build文件夹的所有文件,然后再重新执行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,因为当时真的被搞得头大

欢迎交流指正,如果觉得写得不错请给我点个吧~

 

 

 

评论 8
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值