超详细图文带你手把手安装 CUDA 和 CUDNN

发布时间 2023-09-12 20:16:16作者: 虎虎生威啊

CUDA 安装教程(手把手)

一、简单了解 CUDA 与 CUDNN

1、什么是 CUDA

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

2、什么是 CUDNN

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

3、CUDA 与 CUDNN 的关系

CUDA 看作是一个工作台,上面配有很多工具,如锤子、螺丝刀等。cuDNN 是基于 CUDA 的深度学习 GPU 加速库,有了它才能在 GPU 上完成深度学习的计算。它就相当于工作的工具,比如它就是个扳手。但是 CUDA 这个工作台买来的时候,并没有送扳手。想要在 CUDA 上运行深度神经网络,就要安装 cuDNN,就像你想要拧个螺帽就要把扳手买回来。这样才能使 GPU 进行深度神经网络的工作,工作速度相较 CPU 快很多。

二、CUDA 安装前准备,安装 VS

CUDA 下载连接:
https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64

1、在安装之前一定要先安装 VS,否则在安装 CUDA 过程中就会出现如下情况,从而无法正常安装

2、VS_Community 下载链接:

https://my.visualstudio.com/Downloads?PId=8228

登录自己的账号,最好下载 2017 版本的,实在找不到选择 2019 或者 2022 版本的下载均可,注意一定选择 Community 免费社区版。

3、下载完后,双击安装

4、在安装选项中勾选 “C++ 桌面开发” 即可

5、点击启动,登录自己的账号初始化完毕即可

6、启动完毕,关闭 VS 即可

三、查看 CUDA 版本

1、在搜索框输入 NVIDIA,点击 NVIDIA 控制面板

2、选中 “管理 3D 设置”,点击下方的 “系统信息”

3、点击 “组件” 按钮,查看 CUDA 版本

4、可以看到我这里是 CUDA 12.1

四、下载并安装 CUDA

1、进入 CUDA 官网**,根据上面确定的 CUDA 版本找到对应版本**

2、选择自己的版本,Installer Type 方式选择 exe(local) , 之后点击 “Download” 按钮

3、双击下载的 exe 安装包

4、选择临时解压位置,点击 “OK”,等待解压完毕

5、等待系统检测完毕,点击同意并继续

6、选择自定义安装,点击下一步

7、继续点击 “下一步”

8、选择安装位置,一定要记住,之后需要用到,一般默认即可

9、等待安装完成

10、点击 “下一步”

11、安装完毕,取消勾选,点击关闭即可

12、查看系统环境变量,点击设置—> 关于—> 高级系统设置

13、查看系统环境变量

14、可以看到,CUDA 环境变量已经自动加入到系统中

五、测试 CUDA 是否安装成功

1、win+R   快捷键,之后输入 cmd,调出命令行终端

2、在命令行输入 nvcc -V【返回版本号则安装成功】

六、下载并安装 CUDNN

1、进入 **CUDNN 官网 **,需要注册 / 登录后才可以下载**

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

2、登录后,选择我们与 cuda 对应的版本下载安装即可

3、将下载的压缩包解压

4、将解压后 bin 目录的内容全部放到 CUDA 对应的 bin 目录下

5、将解压后 include 目录下的内容全部放到 CUDA 对应的 include 目录下

6、将解压后 lib 目录下 x64 目录中的内容全部放到 CUDA 对应的 lib 目录下的 x64 目录中

7、打开系统环境变量,步骤同上

8、编辑 Path 变量

9、将这四条添加进去,前两条在之前已经自动添加了,点击 “新建” 添加后两条即可

九、测试 CUDNN 是否安装成功

通过 cmd 打开命令行终端,步骤同上,在终端输入 nvidia-smi,返回 GPU 型号则安装成功

综上,我们便成功安装了 CUDA 以及 CUDNN,后续便可以轻松安装 GPU 版本的深度学习框架,感受 GPU 的速度,陆续跟新,敬请期待!

参考连接 https://mp.weixin.qq.com/s/rht6A0xPkBvRRwf8rH0qsg