Windows下安装Pytorch和CUDA及配置CUDNN教程

本文详细指导了在Windows系统中查看显卡驱动版本,选择和安装CUDA11.6及cuDNNV8.9.3,配置环境变量,最后安装并验证PyTorch使用CUDA的过程。
摘要由CSDN通过智能技术生成

一、查看电脑的显卡驱动版本

1.键盘点击Windows+R或者在搜索栏输入cmd,进入命令提示符,输入nvidia-smi,查看显卡驱动版本

如图所示,本电脑显卡驱动版本为:536.99

二、查找合适的CUDA版本以及cuDNN版本

(一)、查找合适的CUDA版本

1.点击此链接https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html,进入CUDA Toolkit官方文档界面,向下翻阅,找到以下图表:

可以看到本电脑最高可以装12.2版本的CUDA,保险起见,我选择安装11.6版本的

(二)、查找合适的cuDNN版本

1.点击此链接https://developer.nvidia.com/rdp/cudnn-download,进入cuDNN Archive界面下,查找适合CUDA11.6的cuDNN版本

我们cuda是11.6,这里就选择cuDNNV8.9.3版本的for CUDA11.x版本即可三、下载CUDA和cuDNN

三、下载CUDA和cuDNN

前面已经确定需要下载的CUDA版本和cuDNN版本,接下来就是找到它们并进行下载

(一)、下载CUDA11.6
1.点击此链接https://developer.nvidia.com/cuda-downloads,进入CUDA下载界面,发现默认下载CUDA最新版本CUDA 12.4,但这个版本并不适合我们,所以我们需要点击右下角的Archive of Previous CUDA Releases,找到CUDA的早期版本

在CUDA Toolkit Archive界面下,点击CUDA Toolkit 11.6(这是我所需要的CUDA版本)

在CUDA Toolkit 11.6 Download界面,进行如下选择,然后点击Download下载

(二)、下载cuDNN v8.9.3
1.点击此链接https://developer.nvidia.com/rdp/cudnn-download,进入cuDNN版本库,找到cuDNN v8.9.3

点击该项,选择Windows版本

四、安装CUDA11.6配置cuDNN v8.9.3

(一)、安装CUDA

1、一直默认下一步就好了,俗称傻瓜式安装,注意以下几点:

2、没安装VS,将CUDA安装组将中的VS取消勾选:

3、选择安装位置,保持默认路径(记住这个路径,把这个路径保留下来,后面添加环境变量时,会用到),直接点击下一步:

4、安装完成后,添加CUDA的环境变量(如果已经存在,则不需要重复添加)

5、检验CUDA 11.6是否安装成功

键盘点击Windows+R或者搜索栏输入cmd,打开命令提示符

在命令提示符下,输入以下命令

nvcc -V

若可以查看到CUDA版本信息,则说明安装成功:

(二)、配置cuDNN

1.将之前下载的cuDNN压缩文件进行解压,在解压后的文件中找到如下文件将里面对应的bin,lib,include内的文件(注意是文件不是目录)复制粘贴(不是替换)到CUDA安装目录(前面记的那个路径)的各自对应bin,lib,include文件夹下(本人是默认路径:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2)

2、给cuDNN添加相应的环境变量

在系统变量设置界面,选中path,再点击编辑,

点击新建,逐个填入如下环境变量(已经存在的环境变量,不需要重复添加)

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\include
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\lib
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\libnvvp

3、检验是否成功安装:

在cmd命令行输入如下命令进入文件夹:

cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\extras\demo_suite

然后再输入:bandwidthTest.exe

接着输入:deviceQuery.exe

表示安装成功!!!

五、安装Pytorch

1、进入Pytorch官网:Previous PyTorch Versions | PyTorch

找与我们安装的cuda版本一致的下载命令,本文装的是cuda11.6,于是在cmd命令行输入如下命令:

出现下图所示:

2、验证是否安装成功:打开命令行输入python进入交互式编辑,输入以下命令

import torch
x = torch.rand(5,3)
print(x)

验证Pytorch能否使用GPU,导入torch模块

import torch
print(torch.cuda.is_available())

输出True则当前Ptorch能够正常使用CUDA,返回False则说明配置有问题不能正常使用CUDA。

喜欢文章可以点赞收藏,欢迎关注,如有错误请指正!

参考

深度学习环境搭建:Pytorch与CUDA的安装_安装cuda和pytorch-CSDN博客

史上最详细的Pytorch+CUDA+CUDNN的安装(GPU版)_pytorch cuda-CSDN博客

Windows下安装CUDA并配置cuDNN教程_windows安装cudnn-CSDN博客

  • 23
    点赞
  • 23
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值