CUDA(CUDA Toolkit)、显卡算力、PyTorch版本之间的匹配

这篇具有很好参考价值的文章主要介绍了CUDA(CUDA Toolkit)、显卡算力、PyTorch版本之间的匹配。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

刚开始在Anaconda搞环境的时候没注意到这四者之间的对应关系,点进去CUDA Toolkit的安装官网:

CUDA Toolkit 12.1 Downloads | NVIDIA DeveloperResources CUDA Documentation/Release NotesMacOS Tools Training Sample Code Forums Archive of Previous CUDA Releases FAQ Open Source PackagesSubmit a BugTarball and Zip Archive Deliverableshttps://developer.nvidia.com/cuda-downloads就直接下载安装了最新版本的CUDA Toolkit12.1.66,回过神来后,发现显卡版本(算力)、驱动版本(Driver Version)、CUDA Toolkit(CUDA Version)和PyTorch版本这四者之间有一定的匹配关系。简单地说就是:需要按照显卡型号(算力)安装驱动,再根据驱动版本选择相应版本的CUDA Toolkit进行安装(CUDA Toolkit可安装多个版本)。


目录

硬件&系统信息

对照流程

一、查看自己的显卡/GPU型号

二、查看自己的驱动版本(Driver Version)和CUDA版本(CUDA Version)

三、CUDA版本和驱动版本之间的对应

四、CUDA版本和驱动版本不匹配的应对方法

方法1:使用低版本CUDA Toolkit

方法2:更新驱动

五、PyTorch对应的版本


硬件&系统信息

显卡:英伟达GeForce3070Ti

系统:Windows11


对照流程

一、查看自己的显卡/GPU型号

两种方式查看:

1、任务管理器 → 性能(可以看到GPU的型号);或者

2、在开始菜单搜索设备管理器 → “显示适配器”中可以看到GPU的型号(一般是看自己独显,也就是英伟达显卡型号)

此外,官网给出了每一款显卡的算力(可看可不看,主要得知道自己显卡的型号):

https://developer.nvidia.com/zh-cn/cuda-gpushttps://developer.nvidia.com/zh-cn/cuda-gpus

CUDA(CUDA Toolkit)、显卡算力、PyTorch版本之间的匹配

CUDA(CUDA Toolkit)、显卡算力、PyTorch版本之间的匹配

对应什么产品什么型号就自行选择即可。不同型号的显卡算力可能是一样的,如:3070和3070Ti都是8.6的算力。


二、查看自己的驱动版本(Driver Version)和CUDA版本(CUDA Version)

按start+R,输入cmd,打开windows命令行,输入命令:

nvidia-smi

CUDA(CUDA Toolkit)、显卡算力、PyTorch版本之间的匹配

可见我当前的驱动版本是531.14,CUDA版本是12.1。


三、CUDA版本和驱动版本之间的对应

那么CUDA Toolkit12.1.66的需要的驱动版本范围应该在对应的版本的说明文档里标明,如12.1版本的说明在以下官方网站中标出:(我用的是Windows系统)

CUDA Toolkit与驱动版本的对应https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html#notices

下方的链接是官网给出的对照表格,如果你懒得在上面的网站里翻,也可以直接看这个: 

Table 3 CUDA Toolkit and Corresponding Driver Versions

可以看到表格里写的清清楚楚:

CUDA(CUDA Toolkit)、显卡算力、PyTorch版本之间的匹配

如果你想要安装CUDA 11.6 GA,那你的驱动版本起码得是511.23;

如果你的驱动版本已经达到了546.12,那上表里的CUDA你都能安装(CUDA Toolkit是可以同时安装多个版本的,用哪个版本的CUDA就把该版本写入环境变量就可以)。


四、CUDA版本和驱动版本不匹配的应对方法

如果现在你的CUDA版本和驱动版本不匹配,就可以:

  • 方法1:使用低版本CUDA Toolkit;或者
  • 方法2:更新驱动

方法1:使用低版本CUDA Toolkit

关于这个方法涉及多版本CUDA切换,等写完放链接。

方法2:更新驱动

驱动是可以更新的。附上驱动安装/更新的官方网站:

人工智能计算领域的领导者 | NVIDIA

CUDA(CUDA Toolkit)、显卡算力、PyTorch版本之间的匹配

点击“驱动程序”,出现以下界面:

CUDA(CUDA Toolkit)、显卡算力、PyTorch版本之间的匹配

根据你的显卡型号和需求选择即可,点击“搜索”出现驱动下载界面:

CUDA(CUDA Toolkit)、显卡算力、PyTorch版本之间的匹配

点击“安装”即可。可以看到对于GeForce 3070Ti 可用的最新的驱动版本已经更新到546.17了。

更新后的驱动版本也许能够支撑起CUDA版本,但如果更新后还是达不到当前使用的CUDA版本对驱动版本的要求,那就必须换个低版本的CUDA。


五、PyTorch对应的版本

若不知道如何在Anaconda的环境中安装PyTorch,请看:Anaconda+PyCharm+PyTorch+Gym深度强化学习环境搭建 送新手直接送进炼丹炉_pycharm安装gym库_在屏幕前出油的博客-CSDN博客Anaconda+PyCharm+PyTorch+Gym强化学习环境搭建流程,新手强化学习炼丹炉入口。https://blog.csdn.net/qq_43674552/article/details/127343863?spm=1001.2014.3001.5502那么现在来看PyTorch了,目前官网最高支持的是CUDA版本是11.8,有些滞后,和我的CUDA12.1.66版本是不匹配的。但之前我在Anaconda环境中安装的也是CUDA版本为11.8的PyTorch(版本如下图):

CUDA(CUDA Toolkit)、显卡算力、PyTorch版本之间的匹配

 所以,由于PyTorch给出的CUDA的版本有些滞后,就需要验证一下支持CUDA11.8的PyTorch是否和CUDA12.1.66是兼容的,于是我在这个PyTorch官网安装页往下翻了翻:

CUDA(CUDA Toolkit)、显卡算力、PyTorch版本之间的匹配

也没说什么特别的,就是说conda和CUDA要适配你的机器,并且在你的可选范围内,选择CUDA版本越高的PyTorch越好。

接着官网给了安装成功的认证过程:

CUDA(CUDA Toolkit)、显卡算力、PyTorch版本之间的匹配

翻译过来就是说,以下步骤可以帮你认证你的PyTorch是否安装成功,并且你的CUDA是否能为PyTorch所用。

用PyCharm新建一个空项目,环境配置选择安装了PyTorch(CUDA version = 11.8)的那个环境,按照上图官网文档给出的命令验证一下即可:

import torch
x = torch.rand(5, 3)
print(x)
#如果你的输出是类似于官网给出的那个tensor的样子,说明torch可用

torch.cuda.is_available()
#若返回True则说明CUDA可用

CUDA(CUDA Toolkit)、显卡算力、PyTorch版本之间的匹配

 我输出了一个和官网给出的相似的结果(绿框里的那个tensor),并告诉我我的CUDA可以使用,但不知道跑模型的时候会不会提示什么不兼容的错误(应该不会,到时候再说)。

哦对了,CUDA官网还说CUDA是向下兼容的,所以我觉得高版本CUDA会兼容支持低版本CUDA的PyTorch。但如果选择保守一点,就严格按照对应的版本安装使用即可。文章来源地址https://www.toymoban.com/news/detail-450862.html

到了这里,关于CUDA(CUDA Toolkit)、显卡算力、PyTorch版本之间的匹配的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

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

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

相关文章

  • PyTorch + CUDA 版本匹配安装

    目录 1. 问题描述 2. 调查和解决 2.1 初步调查 2.2 官网安装方法 2.3 如果还是不匹配呢?         系统:Windows10,CUDA11.1.96         开始学习PyTorch。在用PyTorch进行一个深度学习训练时发现报告以下Warning信息:         有点纳闷。用Tensorflow-GPU也不是一天半天了,为什

    2023年04月26日
    浏览(52)
  • 解决pytorch和cuda版本不匹配问题

    在跑项目时遇到了以下问题: RuntimeError: CUDA error: no kernel image is available for execution on the device. Anaconda安装pytorch遇到一直循环Solving environment问题 CondaHTTPError:HTTP 000 CONNECTION FAILED for url<https://mirrors.tuna.tsinghua.edu.cn/anaconda*** Executing transaction: failed 是pytorch和电脑自身cuda版本不匹配

    2023年04月08日
    浏览(44)
  • mmcv与cuda,pytorch版本匹配要求

    mmcv与cuda、pytorch版本兼容要求,见mmcv官方文档:https://mmcv.readthedocs.io/zh_CN/latest/get_started/installation.html#pip 安装部分。 目前网页上默认最新版2.x版本,若要切换旧版,点击页面左下角切换即可 。 查看自己的cuda和torch版本: 点击文档链接 选择自己所需版本,拷贝对应的安装命

    2024年02月16日
    浏览(50)
  • GPU版本的pytorch安装(显卡为3060ti,如何选择对应的cuda版本)

    显卡为3060ti g6x,操作系统win10 要清楚下面的几个常识 1.GPU和CPU是采用不同架构设计出来的,简单来说,GPU会比CPU多很多计算单元,用于训练网络时,速度比CPU快很多。 2.CUDA是一个计算平台和编程模型,提供了操作GPU的接口。 3.网上很多教程说的安装CUDA其实是指CUDA Toolkit,是

    2024年02月08日
    浏览(48)
  • Pycharm搭建CUDA,Pytorch教程(匹配版本,安装,搭建全保姆教程)

            最近训练模型跑代码需要用到nvidia的cuda架构加速,结果网上几乎找不到什么能直接解决问题的教程,最后东拼西凑了几个小时才搭建完成,所以想整理出这篇集百家之精华的教程,防止自己以后太久不用忘记了。         首先的先知道配置好这些环境需要准备好哪

    2024年02月04日
    浏览(50)
  • 支持CUDA运算的显卡算力表

    GPUs supported Supported CUDA level of GPU and card. CUDA SDK 1.0 support for compute capability 1.0 – 1.1 (Tesla CUDA SDK 1.1 support for compute capability 1.0 – 1.1+x (Tesla) CUDA SDK 2.0 support for compute capability 1.0 – 1.1+x (Tesla) CUDA SDK 2.1 – 2.3.1 support for compute capability 1.0 – 1.3 (Tesla) CUDA SDK 3.0 – 3.1 support for compu

    2023年04月21日
    浏览(37)
  • Pytorch、CUDA和cuDNN的安装图文详解win11(解决版本匹配问题)

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

    2023年04月18日
    浏览(57)
  • 了解NVIDAI显卡驱动(包括:CUDA、CUDA Driver、CUDA Toolkit、CUDNN、NCVV)

    转载 一篇 背景   开发过程中需要用到GPU时,通常在安装配置GPU的环境过程中遇到问题;CUDA Toolkit和CUDNN版本的对应关系;CUDA和电脑显卡驱动的版本的对应关系;CUDA Toolkit、CUDNN、NCVV是什么呢? 举个例子 安装TensorFlow2.1过程中,想要使用到电脑的显卡来进行开发,但是发现默

    2024年02月13日
    浏览(55)
  • 一张图了解GPU、CUDA、CUDA toolkit和pytorch的关系

    GPU、Cuda Driver和 Cuda Toolkit的图解关系 省流,简略版本 Nvidia Driver驱动 :操作系统和硬件GPU进行沟通交互的程序,没这个驱动,GPU就是一个摆设,调用不起来,所以得先装这个。装了这个电脑就能用GPU了,但是深度学习搞不定。 CUDA Toolkit工具包 :是基于驱动程序,用来实现

    2024年02月01日
    浏览(39)
  • 【CUDA】GPU 算力与 CUDA 版本对应关系

    官方算力表:https://developer.nvidia.com/cuda-gpus#compute 2.1. 信息来源 1 https://docs.nvidia.com/datacenter/tesla/drivers/index.html#cuda-arch-matrix 2.2. 信息来源 2 https://en.wikipedia.org/wiki/CUDA#GPUs_supported

    2024年01月19日
    浏览(35)

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

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

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

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

二维码1

领取红包

二维码2

领红包