windows下caffe GPU版本配置
Posted outthinker
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了windows下caffe GPU版本配置相关的知识,希望对你有一定的参考价值。
由于项目需要,所以在自己本子上配置了一下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就行了;
以上是关于windows下caffe GPU版本配置的主要内容,如果未能解决你的问题,请参考以下文章
windows环境Caffe安装配置步骤(无GPU)及mnist训练