安装之前,先简单了解一下CUDA和cuDNN:

CUDA (ComputeUnified Device Architecture),是显卡厂商NVIDIA推出的运算平台。 CUDA是一种由NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题。

NVIDIA cuDNN是用于深度神经网络的GPU加速库。它强调性能、易用性和低内存开销。NVIDIA cuDNN可以集成到更高级别的机器学习框架中,如谷歌的Tensorflow、加州大学伯克利分校的流行caffe软件。简单的插入式设计可以让开发人员专注于设计和实现神经网络模型,而不是简单调整性能,同时还可以在GPU上实现高性能现代并行计算。

一、CUDA安装

1. 查看本机的CUDA驱动适配版本

  • 桌面空白处右键打开“NVIDIA控制面板”

  • 点击对话框左下方的“系统信息-->组件”

 可以看到本机支持的是CUDA9.1.84版本,表示本机GPU支持9.1.84及以下的版本,不支持更高版本。

2. CUDA版本下载

官网下载地址:

CUDA Toolkit Archive | NVIDIA Developer

 如果要查看以前的版本,点击右下方的“Archive of Previous CUDA Releases”

 根据步骤1查到的本机CUDA版本选择合适的安装版本,

选择与本机操作系统一致的CUDA版本。

选择后发现对应的版本cuda_9.1.85_win10.exe高于本机CUDA9.1.84版本,为了兼容性,较低版本重新选择

 完成下载。

3. 安装CUDA

 

点击上图图标,打开安装程序

安装路径可以默认也可以自定义

 点击OK

 点击“同意并继续(A)”

 如果不知道怎么选择,就选“精简安装”

 然后,耐心等待……

 

 CUDA安装完成后,打开命令行输入nvcc -V, 成功的话会返回cuda版本号;

 总结找到所下载的CUDA,直接安装即可。如果想自定义的话,记住所选择的安装路径。

 二、 cuDNN安装

  1. cuDNN版本下载

官网下载地址:cuDNN Archive | NVIDIA Developer

cuDNN Archive | NVIDIA Developer

 

 

下载cuDNN是需要登录英伟达开发者账户的,注册一个并填写问卷就行了,很简单。

注意:必须选择和所安装CUDA匹配的版本。

选择合适的版本,然后选择Windows10系统,开始下载。

5. 安装cuDNN

解压cuDNN安装压缩包,

 可以看到bin、include、lib目录

 将 bin 目录里面的内容复制,然后粘贴到  C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA   对应的  bin 目录里面:

 同样的,include 和 lib 目录里面的文件也复制到相应的文件夹里面。

6. 添加环境变量

注:这一步很重要!

桌面选择“此电脑”,鼠标右键单击,选择“属性”

打开系统设置界面

 

 选择右侧“高级系统设置”

按照下图检查环境变量(包括用户变量和系统变量),如有遗漏,按图进行添加。

注意版本号要按照自己下载的版本写。

用户变量Path中添加:

%CUDA_LIB_PATH%;%CUDA_BIN_PATH%;%CUDA_SDK_LIB_PATH%;%CUDA_SDK_BIN_PATH%;

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin

C:\Program Files\NVIDIA GPU Computing Tookit\CUDA\v9.0\libnvvp

C:\ProgramData\NVIDIA Corporation\CUDA Samples\v9.0\common\lib\x64

C:\ProgramData\NVIDIA Corporation\CUDA Samples\v9.0\bin\win64

C:\Program Files\NVIDIA Corporation\NVSMI

系统变量中对比以下变量,若有缺失,手动添加。

 下方红框内标注的变量一般是手动添加的。

7. 查看安装结果

有两种方式可以验证cudnn是否安装成功。

方式一:Win+R打开终端,输入nvidia-smi

若能看到GPU型号,则表明驱动安装成功。 

方式二:

Win+R启动cmd,cd到安装目录下 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\extras\demo_suite,然后分别执行bandwidthTest.exe 和 deviceQuery.exe

 

 看到“PASS”即表示cudnn安装成功。

Logo

瓜分20万奖金 获得内推名额 丰厚实物奖励 易参与易上手

更多推荐