Windows 10下CUDA及cuDNN的安装

为了配置OPenPose的环境,所以先得安装好CUDA和cuDNN。CUDA是NVIDIA推出的用于自家GPU的并行计算框架,CUDA在NVIDIA的GPU上运行,而且只有当要解决的计算问题是可以大量并行计算的时候才能发挥CUDA的作用。cuDNN(CUDA Deep Neural Network library):是NVIDIA打造的针对深度神经网络的加速库,是一个用于深层神经网络的GPU加速库。如果你要用GPU训练模型,cuDNN不是必须的,但是一般会采用这个加速库。

好了,首先下载CUDA和cuDNN,官网链接如下:

CUDA官网下载地址:https://developer.nvidia.com/cuda-downloads

cuDNN官网下载地址:https://developer.nvidia.com/cudnn

注意下载cuDNN应该是与下载的CUDA对应的版本。我下载的是CUDA v9.2,然后cuDNN是v7.2.1。下载后就可以安装了。

安装CUDA

安装CUDA之前必须安装好Visual Studio,不然还得再安装。我装的是Visual Studio 2017,装好VS之后可按照提示直接安装CUDA即可。

注意:在安装完成之后,你可能会找不到你自定义路径下安装的CUDA。这是因为之前选择的是临时解压路径,不修改的话是默认的C盘/tmp文件夹,安装时候解压完目录会被自动删除。CUDA会自动强制默认安装在C盘。可以在 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA找到。

安装cuDNN

官网下载好后可以直接解压文件夹。然后将这个文件夹下的文件按照如下操作复制到CUDA路径下:

  1. Copy  <installpath>\cuda\bin\cudnn64_7.dll     to    C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin.
  2. Copy  <installpath>\cuda\ include\cudnn.h    to    C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\include.
  3. Copy  <installpath>\cuda\lib\x64\cudnn.lib    to    C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib\x64.

 然后配置环境变量。在cmd下输入sysdm.cpl,然后选择高级,再选择环境变量。确定下面系统变量的设置:

Variable Name: CUDA_PATH 
Variable Value: C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2(根据自己的CUDA)

在VisualStudio项目中包含cudnn.lib。打开VisualStudio项目并右击项目名称。单击 Linker > Input > Additional Dependencies.
添加cudnn.lib并单击OK。

哈哈哈,就此告一段落。其实官网有详尽的安装教程:https://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html#install-windows

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值