win10安装cuda+cudnn

305 阅读2分钟

本文主要说明CUDA和Cudnn是什么,然后以图文结合的方式,详细记录了在win10笔记本上的安装cuda和cudnn软件包的步骤,供大家参考。

一、CUDA和cuDNN简述

1、CUDA

按百度百科定义,CUDA是英伟达推出的一个通用并行计算架构,是为提高GPU计算效率的软件工具箱。

2、cuDNN

cuDNN是英伟达为深度神经网络打造的一个GPU加速库,使得模型训练推理时速度更快​,不是必须的,类似在自行车上装一个电动马达,使其速度更快。

二、CUDA和cuDNN安装

1、CUDA安装

(1)查看电脑gpu显卡驱动版本号

鼠标右键--》选择NVIDIA控制面板,打开NVIDIA控制面板

image.png

image.png 然后选择:帮助--》系统信息--》组件,即可看到显卡驱动版本信息,小编电脑的版本号为11.7.101

image.png

image.png (2)根据显卡驱动版本号下载对应的CUDA版本

建议CUDA版本小于等于显卡驱动版本号,CUDA官网下载地址:developer.nvidia.com/cuda-toolki…

image.png 由于小编的电脑显卡驱动版本为11.7.101,同时为节约电脑硬盘空间,选择下载安装CUDA Toolkit 11.7.0版本。其中Version是指操作系统win10

image.png (3)开始安装

下载完成后,双击exe文件,然后选择临时解压目录,然后开始解压。

image.png

image.png

在安装选项这里,建议选择自定义安装。

image.png 勾选复选框 image.png 选择下一步 image.png 安装完成点击关闭 image.png (3)验证安装是否成功

打开命令行工具,输入nvcc -V查看cuda是否成功,当出现以下信息时表示CUDA安装成功。

image.png

2、cuDNN安装

小编电脑安装CUDA10.0,因此下载对应的cuDNN版本,下载地址: developer.nvidia.com/rdp/cudnn-a…

image.png

下载完后解压,解压后的文件如下

image.png

然后把解压后的所有文件拷贝安装CUDA的根目录

(小编目录为C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.7)下面即可。

image.png 至此,安装完毕。