Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

这篇具有很好参考价值的文章主要介绍了Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

本文已收录于Pytorch系列专栏: Pytorch入门与实践 专栏旨在详解Pytorch,精炼地总结重点,面向入门学习者,掌握Pytorch框架,为数据分析,机器学习及深度学习的代码能力打下坚实的基础。免费订阅,持续更新。

可能出现的问题:

  • CUDA和cuDNN版本不匹配
  • CUDA和Pytorch版本不匹配
  • cuDNN和Pytorch版本不匹配
  • 显卡不支持CUDA该版本
  • 已经装完部分,发现版本不匹配准备卸载。

说在前面的话!

  1. 在ubuntu系统下,可以尝试装多个cuda版本,然后通过conda安装对应的Pytorch版本。通过软连接的方式来实现cuda版本的切换。**但是,在win系统下,最好是用相同的支持版本,以免不匹配。**不用纠结是否向下兼容等等问题,最优的方法就是安装相同的版本。
  2. 对于CUDA的版本,我推荐用以往的稳定版本,就是指目前还在一直维护的比较旧的版本,原因有很多:
    • 比较旧的版本有强大的社区支持,可以方便地找到前人总结地bug解决方案,而不是遇到最新问题时能力不够导致的一筹莫展。
    • 较旧的版本至今仍在维护,说明其仍有很大的价值,用户基数很多,能确保开发的流畅与稳定。
    • 由于学术界和工业界都喜欢用比较稳定的版本来搭建模型,因此如果想要复现论文,或是pull别人的代码修改,较新的版本很有可能会出现错误。

CUDA的安装

1.查询支持的最高版本

首先安装之前要先检查我们显卡所支持的最高的CUDA版本:

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

目前是11.6的驱动,因此我的显卡最高是可以支持到CUDA11.6版本的。

知道了我们的最高支持版本之后,我们就可以在小于等于该版本的CUDA中选择了。

2.查询Pytoch与cuDNN版本

首先不用着急挑选CUDA的版本。我们先看下pytorch以及cuDNN的版本支持情况。

Pytorch:https://pytorch.org/

cuDNN:https://developer.nvidia.com/rdp/cudnn-archive

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

可以看到对于win系统,Pytorch支持的版本有10.2,11.3,11.6等。

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

cuDNN支持的版本有10.2,11.5,11.4等等。

这里我推荐装10.2或者11.3,原因见开头,不多做赘述。这里以11.3为例。

3.下载CUDA

在CUDAhttps://developer.nvidia.com/cuda-toolkit-archive中,寻找CUDA Toolkit 11.3版本,然后寻找相应的版本下载即可。

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

4.安装CUDA

安装CUDA,首先需要选择CUDA的临时解压路径,这个临时解压文件夹会在安装完成后自动删除,这里建议默认。

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

解压完成后会有安装程序,同意即可。接下来的安装选项选择自定义:

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

在安装CUDA中取消这个VS有关的组件:

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

底下这三个也没必要,可安可不安,看个人选择:

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

安装路径仍然建议默认,在Program Files中,方便以后寻找。建议记住这里的CUDA路径。

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

然后一直确认最后关闭即可,这里不多赘述。

5.验证CUDA是否安装成功

我们在cmd中使用cd命令切换到刚刚CUDA的安装路径下的bin(二进制)文件夹下,再执行nvcc -V命令。

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

可见CUDA是正确安装的。

cuDNN的安装

在cuDNN的版本中,选择支持该版本的CUDA即可,这里我们看到v8.5.0的cuDNN支持CUDA 11.X,说明兼容cuda11.x全系列。点击下载即可。

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

接下来,解压该压缩包,然后复制其中的文件夹

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

粘贴到CUDA的安装目录下,即完成了cuDNN的安装。

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

验证是否安装成功

在cmd中进入到demo文件夹:路径为C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\extras\demo_suite

执行bandwidthTest.exe,如果运行结果出现了PASS即代表安装成功。

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

再输入命令deviceQuery.exe查询设备。

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

这里会显示你的GPU型号,以及PASS,表示CUDA和cuDNN都安装成功了。

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

Pytorch安装

进入Pytorch官网https://pytorch.org/,选择需要安装的pytorch版本。这里安装方式可以选择pip。可以看到有生成的command,里面有个网站,只需要进入该网址下载即可。

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

下载torch

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

由于我们安装的torch版本为Stable1.12.1,因此我们需要查找前缀为:torch-1.12.1的文件。注意找的是GPU版本,cuxxx代表CUDA版本xxx。这里我们找到对应的cu113,然后点击下载。cp代表python版本,这里我们选择cp37版本的win下载。

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

然后返回,进入torchvision。

下载torchvision

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

torchvision的版本选择最新就好,但是要与cuda及python匹配,这里直接搜索cu113-cp37

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

下载对应的版本即可。

新建pycharm创建项目,新建项目名称为PytorchTest, 文件名为HelloPytorch。输入以下代码测试

import torch

print("Hello Pytorch{}".format(torch.__version__))

结果报错,ModuleNotFoundError: No module named 'torch',因为当前环境没有torch。

我们可以用conda创建虚拟环境并安装torch。

首先再Terminal中输入conda create -n pytorch-gpu python=3.7,这里python版本与我们要安装的版本相同。

出现Proceed ([y]/n)? 直接输入y即可。

接着输入conda activate pytorch_gpu激活环境。

注意:进入conda虚拟环境后venv前面的提示会变成你的环境名称,如下:

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

如果没有显示,则可能因为pycharm终端采用的是PowerShell。需要在设置中切换。

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

换成如图所示的cmd终端即可。

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

进入终端后切换到下载刚刚torch和torchvision的文件夹中

cd D:\Develop\pytorch_install //因人而异,cd到你的下载torch和torchvision的文件夹中即可。

用pip安装torch

pip install "torch-1.12.1+cu113-cp37-cp37m-win_amd64.whl"

用pip安装torchvision

pip install "torchvision-0.13.1+cu113-cp37-cp37m-win_amd64.whl"

完成安装

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

安装完成后,需要绑定该项目的解释器为这个虚拟环境。因此需要设置:

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

找到该项目的Python解释器,然后点击齿轮,选择add:

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

选择conda,找到已经存在的环境:

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

查找你的anaconda安装的目录

这里可以搜索anaconda并打开文件位置。

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

然后继续右键打开文件位置即可。

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

打开后找到envs(环境)文件夹。

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

找到刚刚创建的环境,复制文件路径到pycharm。

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

在pycharm中选择该文件路径下的python.exe解释器即可

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

然后一直ok,等待python解释器的重载即可,可能需要等一分钟。

然后重新尝试测试代码并运行。

import torch

print("Hello Pytorch{}".format(torch.__version__))

print(torch.cuda.is_available())

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

返回true说明安装成功。

CUDA的卸载

首先,搜索控制面板并打开

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

找到程序卸载

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

可以在列表中找到有关于NVIDIA的相关组件,找到有关于CUDA的组件并卸载即可,其他的可以保留,因为高于该版本的CUDA会更新其他组件的。(本质上临近时间安装的都能卸载

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

右键 -> 卸载 然后在卸载程序中卸载即可。剩下的方法类似。

Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

最后不放心的话可以用火绒等软件清理一下注册表。文章来源地址https://www.toymoban.com/news/detail-417544.html

到了这里,关于Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

  • GTX1060+win10+CUDA11.3+cudnn8.2+pytorch1.11.0——个人配置踩坑日记

    以下是我亲测有效的使用 GTX 1060 的各部分安装版本 电脑系统:window 10 python版本:3.8.13 pytorch版本:1.11.0 CUDA版本:11.3 cuDNN版本:8.2 以下是我个人安装过程的步骤,以及我踩过的坑,这里解决踩坑问题主要参考: RTX3060+win10+CUDA11.2+cudnn8.2.0+pytorch1.8.0 环境——个人配置经验 Wi

    2024年02月06日
    浏览(75)
  • Win11基于WSL2安装CUDA、cuDNN和TensorRT(2023-03-01)

    如果有小伙伴看了2023-03-05更新,发现设置环境变量后运行cuda代码在链接过程中仍然会有报错问题啥的,那我这里建议,先别管2023-03-05更新的内容了,还是按照我博客里的安装步骤一步一步往下安装,用 sudo apt install nvidia-cuda-toolkit 命令去安装nvcc,也不需要配置环境变量了。

    2024年02月02日
    浏览(54)
  • 【Ubuntu 20.04LTS系统】安装CUDA11.8、cuDNN,可进行CUDA版本切换

    https://mirrors.tuna.tsinghua.edu.cn/help/ubuntu/ 更新软件列表和必要的依赖项 步骤一: 下载CUDA安装包 进行CUDA和cuDNN的选择,也可以直接根据官方推荐进行下载安装。 从Nvidia官网下载CUDA https://developer.nvidia.com/cuda-downloads 下方链接,选择更多版本 https://developer.nvidia.com/cuda-toolkit-archiv

    2024年02月16日
    浏览(70)
  • 分享:win10使用 python 让 NVIDA GeForce MX250 显卡发挥余热,安装CUDA,cuDNN和PyTorch

    打开NVIDA更新驱动的官网地址 根据下图的选择,记得 Windows驱动程序类型 要选 标准 ,如图 点击搜索,下面就会列出一大堆的历史驱动,选择第一个,也就是最新的,当前是472.12,点开,再点下载,再点同意并下载,建议用EagleGet 下载完毕后,双击安装,会提示选择一个解压

    2024年01月22日
    浏览(56)
  • 【Win 11】Pytorch-CUDA版 安装指南

    笔者在很久之前就装过Pytorch,但当时装的是CPU版本,今天尝试装GPU版本,几经波折,总结一些问题在此,以少走弯路。 1. 查看自己的CUDA版本 对于英伟达30系显卡,算力达到8.x,一般需要适配11.x的CUDA。自己可以针对性根据自己的显卡算力查看适配的CUDA。 按下WIN+R键,输入

    2024年02月04日
    浏览(38)
  • pytorch的安装(CUDA10.2+cuDNN8.3.3+torch1.10+​torchvision​0.11.1+python3.9)

    (已存网盘和硬盘,文件夹含三个文件) 本文基本逻辑是: 一、先根据电脑硬件的条件获取本身CUDA版本,据此以及表格比较得出cuDNN、torch、torchvision、python版本。 二、在 NVIDIA 官网下载CUDA和cuDNN,获取torch的下载链接,网页提供python3.9的下载链接 三、安装CUDA后,把cuDNN这个

    2024年01月17日
    浏览(84)
  • cuda11.2版本的对应安装的pytorch版本

    因为电脑里有配置paddle环境,当时用的cuda11.2,现在又要配置torch环境,查看torch官网后发现没有cuda11.2版本对应的torch下载。 考虑到版本向下兼容,可能不一定非要下载cuda=11.2对应的那个版本的torch,或许低于这个版本就可以。所以我就选择下载cuda11.1的版本。 以下是pytorch对

    2024年02月12日
    浏览(53)
  • pytorch环境配置(装cuda、cudnn)win10+cuda10.1+cudnn7.6.5+torch1.7.1 && 集显装pytorch

    为了装这个走了太多坑了,所以想写一篇具体教程,有缘人看吧,希望能解决你的问题。(第一次写文章啥也不懂,万一冒犯了啥,麻烦告知我改) 我anaconda很早就装过了,所以这里就不细说了。 电脑配置:win10+1050显卡(很久前买的电脑的) 一些弯路:现在的pytorch安装是可

    2024年02月04日
    浏览(56)
  • 最新win11配置cuda以及cudnn补丁教程

    1、首先使用指令 nvidia-smi 查看电脑支持的** 最高cuda **版本,例如: 本机 12.2 2、进入CUDA下载cuda安装包 https://developer.nvidia.com/cuda-toolkit-archive 2、点击 上方绿色的链接 ,按照图中序号选择的即可,最后点击下载。 **安装时候展开CUDA,后取消勾选图中内容之后,一直next即可*

    2024年04月22日
    浏览(31)
  • Python&aconda系列:GPU深度学习环境搭建:Win11+CUDA 11.7+Pytorch1.12.1+Anaconda以及对应版本

    官方推荐的cuda版本为10.2和11.3,这两种 cuda 支持大多数的 pytorch 版本。 以下是Pytorch和CUDA对应的版本 CUDA 环境 PyTorch 版本 9.2 0.4.1、1.2.0、1.4.0、1.5.0(1)、1.6.0、1.7.0(1) 10.0 1.2.0、1.1.0、1.0.0(1) 10.1 1.4.0、1.5.0(1)、1.6.0、1.7.0(1) 10.2 1.5.0(1)、1.6.0、1.7.0(1)、1.8.0(1)、1.9.0、1.9.0、1.10.0、1.

    2024年02月02日
    浏览(85)

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包