您的位置:首页 > Web前端

Win7 配置caffe框架(VS2013 CUDA7.5 Win764位)

2016-11-14 17:06 537 查看
仅CPU版本参考以下链接即可,感谢链接作者,也感谢该方法的提供者 卜居。
http://blog.csdn.net/d5224/article/details/51915290
若用GPU方法得重新生成解决方案,并应该加入include lib 库,并改变CommonSettings.props参数,不然会报错。亲测有效。

编译器是:release x64(默认)

具体:1、配置vs环境

vs里选定项目  上面的目录栏  视图->属性管理器

然后把每个目录下的release|x64 下的CommonSettings里的VC++目录里的 包含目录和库目录 改为 你的cuda路径的include和lib

我的在这目录下 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v7.5\include

                            C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v7.5\lib\x64

哦,对了  bin 要加入系统环境变量:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v7.5\bin(一般安装完都会自动加)

然后把cuDNN中的文件放入cuda对应地方。

2、修改CommonSettings.props

        <CpuOnlyBuild>false</CpuOnlyBuild>

        <UseCuDNN>true</UseCuDNN>

        <CudaVersion>7.5</CudaVersion>

        <!-- CuDNN 3 and 4 are supported -->

        <CuDnnPath>D:\caffe\caffe Cai</CuDnnPath> //改为你的cuDNN所在路径

        <ScriptsDir>$(SolutionDir)\scripts</ScriptsDir>

然后 重新生成解决方案。

3、修改lenet_solver.prototxt

最后一行改为 GPU

4、命令框内:

cd切换目录到caffe-master 然后执行命令:

  Build\x64\Release\caffe.exe train -solver examples\mnist\lenet_solver.prototxt

注意:设置好GPU后 想切换成CPU第三步直接改为CPU就好了,但是CPU改为GPU得重新编译。但是,设置好GPU设置后,GPU和CPU可以无缝切换。

错误提示:


CUDA driver version is insufficient for CUDA runtime version

解决方法:下个驱动精灵更新显卡驱动就好~

差不多先这样,笔者先做个笔记。如有问题的童鞋,欢迎留言讨论。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息