欢迎订阅我的YouTube频道,https://www.youtube.com/c/xugaoxiang

windows 10安装CUDA和cuDNN

人工智能 迷途小书童 0评论

软硬件环境

  • windows 10 64bit
  • GTX 1660
  • CUDA 10.2
  • cuDNN 7.6.5
  • visual studio 2019

视频看这里

安装CUDA

首先到https://developer.nvidia.com/cuda-downloads?下载目前的最新版10.2,双击exe文件开始安装

这是一个临时目录,随意选取一个即可

到此cuda就安装好了,接下来我们打开sample目录,通过执行其中的deviceQuery来检查下安装结果

我们选择用visual studio 2019版本打开这个工程,直接编译运行

从结果中可以看出,cuda已经安装成功了

安装cuDNN

cuDNNnvidia公司开发的一套用于DNN也就是深度神经网络的加速库,下面我们就来安装它。进入官网下载安装包,这里需要登录,如果没有账号的话就去注册一个。

下载下来的文件是一个压缩包,解压后有三个文件夹binincludelib,依次将三个文件夹中的内容拷贝到cuda的安装目录(默认情况下是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2)下的binincludelib下,这个要一一对应,如下图所示。

至此,cuDNN也就安装好了。

参考资料

喜欢 (1)
发表我的评论
取消评论
表情

Hi,您需要填写昵称和邮箱!

  • 昵称 (必填)
  • 邮箱 (必填)
  • 网址