一、CUDA和cuDNN简述
1、CUDA
按百度百科定义,CUDA是英伟达推出的一个通用并行计算架构,是为提高GPU计算效率的软件工具箱。
2、cuDNN
cuDNN是英伟达为深度神经网络打造的一个GPU加速库,使得模型训练推理时速度更快,不是必须的,类似在自行车上装一个电动马达,使得骑得更快。
二、CUDA和cuDNN安装
1、CUDA安装
(1)查看电脑gpu显卡驱动版本号
鼠标右键--》选择NVIDIA控制面板,打开NVIDIA控制面板
然后选择:帮助--》系统信息--》组件,即可看到显卡驱动版本信息,小编电脑的版本号为11.7.101
(2)根据显卡驱动版本号下载对应的CUDA版本
建议CUDA版本小于等于显卡驱动版本号,CUDA官网下载地址:https://developer.nvidia.com/cuda-toolkit-archive
由于小编的电脑显卡驱动版本为11.7.101,同时为节约电脑硬盘空间,选择下载安装CUDA Toolkit 11.7.0版本。其中Version是指操作系统win10
(3)开始安装
下载完成后,双击exe文件,然后选择临时解压目录,然后开始解压。
在安装选项这里,建议选择自定义安装
勾选复选框
(3)验证安装是否成功
打开命令行工具,输入nvcc -V查看cuda是否成功,当出现以下信息时表示CUDA安装成功。
2、cuDNN安装
小编电脑安装CUDA10.0,因此下载对应的cuDNN版本,下载地址: https://developer.nvidia.com/rdp/cudnn-archive
下载完后解压,解压后的文件如下
然后把解压后的所有文件拷贝安装CUDA的根目录
(小编目录为C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.7)下面即可。
到此,以成功安装好CUDA+cuDNN。