2021.08.26学习内容 Win10+GeForce GTX1650安装NVIDIA显卡驱动及CUDA11.4+cuDNN8.2

之前主要使用Ubuntu系统,但是个人笔记本更多使用windows,为了方便跑一些pytorch的小代码,所以想在windows配置一下相关环境,达到调用GPU运算的目的。

记录也是为了自己以后有安装需求少走弯路。

我的系统是win10,要调用GPU,首先要安装驱动软件,驱动软件根据GPU(也叫显卡)
版本选择,我的显卡查询结果是GeForce GTX 1650,根据显卡版本再去查询合适的驱动软件,根据查询结果选了一个合适的进行安装,装完后就可以驱动GPU了。但是要在pytorch环境中调用GPU还不行,需要在驱动软件的基础上安装应用程序,可以理解为与pytorch建立联系的工具包或者库文件,也就是CUDA,在英伟达官网中CUDA就是命名为CUDA ToolKit的软件,而对应的cudnn则是用来辅助深度神经网络计算用的,用来跑深度学习的话基本上要安装这两个。根据驱动软件查询合适的应用程序CUDA Toolkit和cudnn,完成安装,一般就可以调用GPU了。具体操作过程如下:

第一步,查询电脑显卡驱动信息,个人电脑显卡查询如下
在这里插入图片描述

第二步,若不确定下什么版本驱动,查询网站:https://www.nvidia.cn/Download/Find.aspx?lang=cn
把个人电脑显卡配置填上
在这里插入图片描述

如果你知道所需安装的环境的CUDA版本,可以根据cuda去找合适的显卡驱动,CUDA与显卡驱动对应关系如下:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html#title-resolved-issues

在这里插入图片描述

第三步,选中符合显卡GPU的驱动下载安装,我基本上是默认的,自己设置都行,影响不大。安装完成进一步安装CUDA,但是CUDA需要对应版本才能调用显卡GPU,所以还要明确CUDA版本。版本可以通过已经安装的显卡驱动查询,右键电脑桌面空白处,打开nvidia显卡驱动面板,如下操作:
在这里插入图片描述
即可获得需要的CUDA版本信息,这是再去下载,基本可以确定不会下错版本,CUDA在NVIDIA官网也叫 cuda toolkit,下载网址如下:https://developer.nvidia.com/cuda-downloads
在这里插入图片描述
点击旧版查询入口如下,相关操作设置如下:
在这里插入图片描述
在这里插入图片描述
下载好CUDA toolkit,下一步就是安装cudnn,网址:https://developer.nvidia.com/rdp/cudnn-download
在这里插入图片描述
在这里插入图片描述

CUDA安装完(建议默认路径),cudnn下载后解压,拷贝三个文件夹至CUDA的如下安装目录覆盖,即可完成安装,
在这里插入图片描述
后续如果没有自动添加环境变量,需要自行添加,设置位置如下。
在这里插入图片描述
都安装完成后,就可以测试是否可以调用GPU了,测试代码如下,最好是包含pytorch环境的终端,结果是True就没有问题。

(pytorch) D:\>python
Python 3.7.11 (default, Jul 27 2021, 09:42:29) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> print(torch.cuda.is_available())
True
>>>

在这里插入图片描述

  • 35
    点赞
  • 136
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

清梦枕星河~

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值