环境配置6-Win10下安装CUDA和cuDNN,配置caffe

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/sinat_30440627/article/details/80840388

到官网下载相应的安装包,cuDNN为lib库,需要先注册NVIDIA账号(或者链接复制到迅雷下载)。

https://developer.nvidia.com/rdp/cudnn-archive

https://developer.nvidia.com/cuda-80-ga2-download-archive




1-cuda下载好后,直接按照默认选项安装。出现下面的提示直接忽略。


2-cudnn下载后解压(比如解压后为C:\cuda\),会有bin、include、lib三个文件夹,将这三个文件夹下的内容复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0下对应的文件夹中。

(以下3、4、5步骤是否还需要做待验证,可以先不做,安装caffe后再观察)

3-添加cudnn环境变量。

    将C:\cuda\bin添加到系统环境变量。

    右击我的电脑——属性——高级——环境变量——系统变量——PATH

4-打开visual studio项目

    选择release、x64

    调试——属性


VC++Directories——Include Directories中添加C:\cuda\include

VC++Directories——Library Directories中添加C:\cuda\lib\x64


Linker——Input——Additional Dependencies中添加cudnn.lib



接下来可以安装caffe

1-下载Windows下的caffe

https://github.com/BVLC/caffe/tree/windows

2-解压出来,用notepad++打开

caffe——scripts——build_win.cmd

3-if DEFINED APPVEYOR不用管,直接看else后的。

MSVC_VERSION=14  (vs2015)

WITH_NINJA=0

CPU_ONLY=0  (如果用GPU)

PYTHON_VERSION=2  (根据安装的python版本)

其余的不用改变,保存后退出。

4-双击运行build_win.cmd开始编译,等待十几分钟。


5-编译成功。将生成的build——lib——release文件夹下的10个文件,拷贝到程序里win_lib下替换。

猜你喜欢

转载自blog.csdn.net/sinat_30440627/article/details/80840388