当前位置:   article > 正文

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

cudnn tarball

在这里插入图片描述

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

目录


操作系统: windows 11
更新时间: 2024年2月21日
安装前提: 电脑必须要有 NVIDIA 独立显卡,且已经安装显卡驱动。


Ⅰ. 前言

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
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11

Ⅱ. 查看显卡驱动兼容性

  • GPU信息、驱动版本可以通过cmd输入nvidia-smi指令来查看。
  • 显卡驱动版本向下兼容,其决定了可安装的CUDA Toolkit的最高版本,高版本的驱动支持低版本的CUDA。
    https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
    
    • 1
  • 比如我的显卡驱动版本是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
  • 1

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
  • 1
  • 2
  • 3
  • 4

出现以下画面则代表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
  • 1
  • 2
  • 3
  • 我电脑是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
  • 1

5.2 CUDA安装

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

  • 选择解压目录

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

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

  • 修改安装位置文件夹

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/酷酷是懒虫/article/detail/810946
推荐阅读
相关标签
  

闽ICP备14008679号