深度学习环境配置——Windows安装CUDA与CUDNN

tyyhmtyyhm

已于 2024-03-22 08:52:05 修改

阅读量5k

点赞数 74

分类专栏：深度学习环境配置文章标签：深度学习 windows 人工智能

于 2024-03-19 13:47:10 首次发布

本文链接：https://blog.csdn.net/tyyhmtyyhm/article/details/136840339

版权

深度学习环境配置专栏收录该内容

5 篇文章 1 订阅

订阅专栏

在这里插入图片描述

关注微信公众号：XRobotSpace
关注微信公众号：依法编程

Ⅰ. 前言

CUDA的安装需要两个包：CUDA和CUDNN（CUDA的一个补丁，用于优化深度学习），先安装CUDA，后安装CUDNN。

CUDA工具包发行说明：
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

CUDA版本汇总：
https://developer.nvidia.com/cuda-toolkit-archive

CUDA安装官方教程：
https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html

CUDNN安装官方教程：
https://docs.nvidia.com/deeplearning/cudnn/installation/windows.html

Ⅱ. 查看显卡驱动兼容性

GPU信息、驱动版本可以通过cmd输入nvidia-smi指令来查看。
显卡驱动版本向下兼容，其决定了可安装的CUDA Toolkit的最高版本，高版本的驱动支持低版本的CUDA。
```
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
```
比如我的显卡驱动版本是529.13，可安装的CUDA最高版本为12.0。（但根据官网的版本兼容性要求，我的显卡驱动版本安装12.3的CUDA是没问题的啊，此处不理解）

Ⅲ. 安装CUDA 12.0

3.1 CUDA下载

下载CUDA Toolkit 12.0.1版本exe文件。

https://developer.nvidia.com/cuda-12-0-1-download-archive

3.2 CUDA安装

在准备安装的盘里面设置两个文件夹：空的解压文件夹Temp（安装后这个文件夹会自动删除，不能将CUDA安装在里面），安装文件夹v12.0，文件夹名自定义。

选择自定义，【精简】会覆盖当前的显卡驱动

Display Driver：我去掉勾选，已经安装了529.13的显卡驱动，不需要重新安装。

修改安装位置文件夹

3.3 查看是否存在CUDA的环境变量

安装完成后会自动添加环境变量。
可以安装多个版本的CUDA，然后修改此处的环境变量决定使用哪个版本。（未测试）

3.4 验证安装是否成功

cmd命令提示符输入“nvcc -V” 或 “nvcc --version”查看安装CUDA的版本号。

nvcc -V
nvcc --version
# 查看环境变量
set CUDA

出现以下画面则代表CUDA安装成功。

Ⅳ. Windows系统安装CUDNN

以CUDA 12.0为例介绍CUDNN安装。

4.1 CUDNN下载

官方安装教程：

https://docs.nvidia.com/deeplearning/cudnn/installation/windows.html

https://developer.nvidia.com/cudnn-downloads?target_os=Windows&target_arch=x86_64&target_version=Agnostic&cuda_version=12

我电脑是Win 11版本，CUDA是12版本，使用Tarball Installation压缩包进行安装。

4.2 CUDNN安装

解压下载的cuDNN压缩包，得到三个文件夹，如下：

复制上面红框中的三个文件夹，并粘贴到CUDA的安装目录下，即刚才装CUDA的位置v12.0文件夹，完成安装。

4.3 检查环境变量配置

CUDA安装完成后会自动添加环境变量，CUDNN无需再次添加。

4.4 验证 `cudnn` 是否安装成功

进入安装目录下的 CUDA\v12.0\extras\demo_suite 文件夹。
鼠标右键单击，选择在终端中打开。
输入 .\deviceQuery.exe 命令，观察窗口输出结果。

继续输入 .\bandwidthTest.exe 命令，两条命令都出现 PASS 说明 cudnn 环境配置成功。

Ⅴ. 安装CUDA 11.8（尝试安装多个版本）

5.1 CUDA下载

下载CUDA Toolkit 11.8.0版本exe文件。

https://developer.nvidia.com/cuda-11-8-0-download-archive?target_os=Windows&target_arch=x86_64&target_version=11&target_type=exe_local

5.2 CUDA安装

在准备安装的盘里面设置两个文件夹：空的解压文件夹Temp（安装后这个文件夹会自动删除，不能将CUDA安装在里面），安装文件夹v11.8，文件夹名自定义。

选择自定义，【精简】会覆盖当前的显卡驱动

Display Driver：我去掉勾选，已经安装了529.13的显卡驱动，不需要重新安装。

修改安装位置文件夹