1. 介绍
- cuda: Compute Unified Device Architecture,是一种有NVIDIA推出的通用并行计算架构, 该架构使GPU能够解决复杂的计算问题。
- cudnn: 是NVIDIA 推出的用于深度神经网络的GPU加速库,他强调性能,易用性和低内存开销。
- cuda和cudnn的关系: cudnn是基于cuda架构开发的专门用于深度神经网络的GPU加速库。cuda可以理解为一个大的商圈,但这个商圈是空的,还未装修。cudnn可以理解为装修后的房间,例如负一楼专门针对游乐(深度神经网络)装修成大型游乐厂。
2. cuda
2.1 选择cuda版本
- 由于不同的GPU驱动程序会有不同的cuda版本,通过以下操作查看自己的驱动版本
- 根据这个链接查看自己对应的cuda版本:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
2.2 下载链接
- 这里是cuda的官网下载链接:https://developer.nvidia.com/cuda-toolkit-archivet
- 按照以下选择,选择自己的版本。如果下载速度较慢,可以复制下载链接,用迅雷下载,2.4GB两分钟就下载好了
2.3 安装cuda 11
-
打开安装文件,会有如下提示,这个是压缩包提取的暂存的文件夹
-
同意并继续
-
自定义安装
-
全部选择,下一步
-
可以自定义选择安装位置,但记得一定要记住自己的安装位置,因为一会儿在配置环境变量的时候会用到。
-
如果哦电脑上没有安装Visual Studio会有这个提示,建议安装Visual Studio,因为不安装的话,cuda的一些功能可能用不了。这里是Visual Studio传送门
-
下一步
-
点关闭就可以了
2.4 检查以及配置环境变量
- 安装完成之后,系统环境变量中,在系统变量中,会有以下新增变量。且在Path这个变量中,也会新增一些变量。
-
在Path中,新增以下系统变量。
1:D:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4
;
2:D:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\bin
;
3:D:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\lib\x64
;
4:D:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\libnvvp
;
-
Win + R
输入cmd
调出命令提示符,输入nvcc -V
查看是否安装成功,安装成功会有以下安装的版本信息。
3 cudnn
3.1 下载cudnn
- 需要注册一个账号,如果没有是不能下载cudnn的。注册大概需要10分钟左右,另外,在注册完毕之后如果无法登陆,建议科学上网后登陆。
- 下载链接:https://developer.nvidia.com/rdp/cudnn-download
- 根据自己安装的cuda版本选择cudnn,这里我们安装的是11.4,所以下载11.4版本
- 如果下载速度过慢,同样可以复制下载链接,使用迅雷加速下载
3.2 解压cudnn
- 解压之后,会有bin,include,lib三个目录
- 把这三个目录的内容,分别拷贝到安装路径对应的目录下。
- 至此,所有,完成,结束。