Windows电脑GPU配置CUDA和cuDNN

前言:最近需要使用电脑的GPU进行并行计算,由于以前也配置过GPU的深度学习环境,但是并没有自己总结如何配置GPU的CUDA和CUDNN,因此趁着这次写了这篇文章,留作记录。

一、清理NVIDIA的CUDA和CUDNN

NVIDIA

  • 从未配置过GPU深度学习环境的话直接跳过
  • 如果以前配置过对应的GPU深度学习环境,需要重新配置的话,计算机中至多保留上面的应用程序,卸载完成后记得用火绒等软件清理无用的注册表

二、安装CUDA Toolkit

1、CUDA版本说明

  • 使用命令nvidia-sminvcc -V均能查看安装的CUDA版本,但不同的是nvidia-smi查看的是显卡驱动程序安装的CUDA版本,nvcc -V是由CUDA Toolkit安装的CUDA版本

image-20220122175005396

image-20220122175022225

像使用GPU进行深度学习通常是指nvcc -VCUDA版本

2、下载CUDA Toolkit安装包

CUDA下载

image-20220122164925694

  • 根据需要下载对应版本的CUDA Toolkit安装包

image-20220122165145110

3、 安装CUDA 11.0

image-20220122164825897

image-20220122165309511

image-20220122165426651
image-20220122165537851

注意:我的电脑上已经安装了最新版本的驱动程序,所以才显示该警告,可不用理会。

image-20220122165620985

image-20220122165638383

注意:精简是安装全部CUDA组件而非最小的组件

image-20220122165805807

注意:只选择CUDA组件且不选Visual Studio Integration,其他组件均不选

image-20220122165934489

注意:CUDA安装位置可根据需要放于不同位置,这里我直接使用默认位置。

image-20220122165958542

image-20220122170336236

image-20220122170347346

4、CUDA安装完成

  • 检查是否安装成功
  • (1) 命令行运行nvcc -V显示对应的版本号表示安装成功

image-20220122170523191

  • (2) 查看环境变量

image-20220122170710595

  • (3) 运行对应程序,均显示PASS则安装成功

image-20220122173224113

image-20220122173326384

三、cuDNN安装

1、下载与CUDA相对应的cuDNN安装包

  • 新用户需要注册后才能下载

image-20220122170943567

  • 同意后,点击Archived cuDNN Releases选择适合的cuDNN版本

image-20220122171028474

  • 选择适合于CUDA版本和平台的cuDNN下载

image-20220122172212722

image-20220122172808484

2、cuDNN安装

  • 将安装包解压后,直接将cuda文件夹下的三个文件夹移入对应的CUDA 安装目录下

我电脑上的CUDA安装目录为C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\

image-20220122173548567

  • 到这里GPU深度学习环境就配置好了,可以愉快地开始学习深度学习了~~~
  • 5
    点赞
  • 32
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值