windows下caffe GPU版本配置

2021-07-19 19:20

阅读:380

由于项目需要,所以在自己本子上配置了一下windows下GPU版本的caffe;

硬件:  win10    ;      gtx1070独显(计算能力6.1)

安装软件:     cudnn-8.0-windows10-x64-v5.1  ;  cuda_8.0.61_win10  ;  NugetPackages.zip  ;  caffe-master

                      可以自己官网下载(我也提供了百度云:链接:https://pan.baidu.com/s/1miDu1qo 密码:w7ja)

参考链接:     https://www.cnblogs.com/king-lps/p/6553378.html(主要按照这篇和下面两篇来配置的)

                       一个word版的配置教程:链接:https://pan.baidu.com/s/1hsEiDEG 密码:d29u

                        http://www.cnblogs.com/zf-blog/p/6139044.html

                       http://blog.csdn.net/tina_ttl/article/details/51722983

                       https://www.2cto.com/kf/201612/575865.html

具体的详细步骤我就先不写了,之前一直在ubuntu上用的caffe,这次配置很轻松,我只说一下几点注意的地方:

(1)安装cuda时,注意   要取消勾选安装driver compoent选项;

(2)cuda安装好后会自动添加到系统变量中,而安装完cuda后最好验证一下(nvcc -V和运行deviceQuery.exe,两者最好都跑一下,后者我运行时出现闪退情况,尝试解决失败,但也没有影响最后的安装);

(3)下载解压完cudnn后,要将解压后得到的三个文件夹合并到cuda相应的文件夹中就OK了;

(4)下载caffe时主要要下载windows版本的caffe,就是文件夹中要有windows这个文件夹;

(5)编译的时候不需要全部编译,编译顺序为libcaffe -> caffe ,如果需要再编译python或者matlab接口;

然后就OK了,试试最简单的mnist就行了;


评论


亲,登录后才可以留言!