PyTorch、CUDA Toolkit 及显卡驱动版本对应关系

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

CUDA驱动及CUDA Toolkit最高对应版本如下:

pytorch和cuda对应版本,深度学习,pytorch,深度学习,python
如果上述没有你想要的,参考官方文档
注:驱动是向下兼容的,其决定了可安装的CUDA Toolkit的最高版本。
CUDA Toolkit版本及可用PyTorch对应关系
pytorch和cuda对应版本,深度学习,pytorch,深度学习,python
注:虽有的卡驱动更新至较新版本,且CUDA Toolkit及PyTorch也可对应更新至新版本。但可能用不了,建议用稳定版
查询驱动版本

nvidia-smi

pytorch和cuda对应版本,深度学习,pytorch,深度学习,python

此处提供三种方法可供选择。
(1)指定CUDA Toolkit版本(推荐)
根据表一查询到可安装的CUDA Toolkit版本,470.141对应最高的CUDA Toolkit版本为11.4。上述表格没有,你可以装11.3的驱动,向下兼容向上不行
运行conda install pytorch cudatoolkit=11.3 -c pytorch即可。
此方法指定CUDA Toolkit版本后,conda会自动匹配到合适版本的PyTorch。

(2)指定PyTorch版本
根据表一查询到可安装的CUDA Toolkit版本,再根据表二查询到合适版本的PyTorch。465对应最高的CUDA Toolkit版本为11.3,11.3可安装PyTorch1.11.0版本(向下兼容)。
运行conda install pytorch=1.11.0 -c pytorch即可。
此方法指定PyTorch版本后,conda会自动匹配到合适版本的CUDA Toolkit。

(3)同时指定CUDA Toolkit版本和PyTorch
根据表一查询到可安装的CUDA Toolkit版本,根据表二查询到合适版本的PyTorch。
运行conda install pytorch=1.11.0 cudatoolkit=11.3 -c pytorch即可。

注:
(1)需要安装torchaudio以及torchvision时将其放在pytorch之后即可,如:conda install pytorch torchaudio torchvision cudatoolkit=11.4 -c pytorch(个人建议这样装)
(2)-c pytorch为采用官网源下载,如果去掉,则在清华镜像源配置后可采用清华镜像源下载,配置清华镜像源可参考:

Linux下conda使用清华镜像源快速安装PyTorch

PyTorch官网
官网下载比较慢
添加清华镜像源:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/

查看添加镜像源:

conda config --set show_channel_urls yes | cat ~/.condarc

显示,则添加成功
pytorch和cuda对应版本,深度学习,pytorch,深度学习,python
在PyTorch官网查看适合版本安装命令
pytorch和cuda对应版本,深度学习,pytorch,深度学习,python

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

注意:不要复制 -c pytorch即可采用清华镜像源下载,否则采用官网源下载。等待安装成功即可。以上可能会出错,则将https改为http,重新添加源即可文章来源地址https://www.toymoban.com/news/detail-826497.html

# 删除已添加的源
conda config --remove-key channels

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

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

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

相关文章

  • NVIDIA显卡驱动及CUDA相关安装流程(包括多版本cuda切换)

    NVIDIA驱动官方下载地址: https://www.nvidia.cn/Download/index.aspx?lang=cn 第一步先查看适合的显卡驱动版本 第二步下载合适的显卡驱动 第三步下载驱动相关依赖 ​ 安装lightdm是在弹窗上选择lightdm[若没有弹窗略过] 第四步禁用nouveau驱动 第五步禁用nouveau内核模块 第六步卸载旧的nvidi

    2024年02月16日
    浏览(48)
  • 【深度学习环境配置】ubuntu 20.04+4060 Ti+CUDA 11.8+pytorch(装机、显卡驱动、CUDA、cudnn、pytorch)

    【深度学习环境配置】ubuntu 20.04+4060 Ti+CUDA 11.8+pytorch(装机、显卡驱动、CUDA、cudnn、pytorch) 📆 安装时间 2023.11.08-2023.11.10 Windows 和 Ubuntu 双系统的安装和卸载 B站教程 【本文基本上跟这个详细教程一致,优先推荐看这个!】ubuntu20.04 下深度学习环境配置 史上最详细教程 【精

    2024年02月04日
    浏览(59)
  • CUDA(CUDA Toolkit)、显卡算力、PyTorch版本之间的匹配

    刚开始在Anaconda搞环境的时候没注意到这四者之间的对应关系,点进去CUDA Toolkit的安装官网: CUDA Toolkit 12.1 Downloads | NVIDIA Developer Resources CUDA Documentation/Release NotesMacOS Tools Training Sample Code Forums Archive of Previous CUDA Releases FAQ Open Source PackagesSubmit a BugTarball and Zip Archive Deliverables h

    2024年02月05日
    浏览(64)
  • Ubuntu 20.04 LTS 系统下 安装Nvidia 显卡驱动、CUDA、cuDNN, 并可进行CUDA版本切换

    因为做深度学习的研究项目,为全新机器在Ubuntu 20.04 LTS 系统下 安装Nvidia 显卡驱动、Cuda、Cudnn。并进行CUDA版本切换 成功安装完成了,写个记录。 步骤一: 安装更新软件列表和依赖项 在安装Nvidia显卡驱动之前,需要更新软件列表和必要的依赖项。 步骤二: 查看GPU型号,并

    2024年02月13日
    浏览(66)
  • 最新版ubuntu22.04安装NVIDIA显卡驱动以及CUDA、CUDNN,和安装驱动gcc版本问题解决。

    1.驱动安装参考下述链接 1.1 NVIDIA显卡驱动、 CUDA 的安装参考这个博主的具体安装步骤,但是最后一步的 cudnn 的安装参考另一个博主,见链接。 1.2 可以选择最新版本的驱动,需要在官网去找最新的版本对应。 2.最新版本我遇到的错误以及解决方法 2.1.在进入tty1界面后ubuntu安装

    2024年04月24日
    浏览(58)
  • 超详细教程——Ubuntu20.04 安装英伟达NVIDIA显卡驱动、CUDA、Cmake以及不同版本的CUDA切换

    在windows和linux双系统下,为ubuntu安装nvidia、cuda、cmake等程序是一个可能会难到各个计算机小白的问题,本文将一步步地带您学会安装以上内容。 一、安装英伟达NVIDIA显卡驱动 对于新安装的linux操作系统,我们需要为它安装nvidia显卡驱动(ubuntu有自带的显卡驱动但更推荐安装

    2024年02月02日
    浏览(54)
  • 【最新】CUDA Toolkit版本及可用PyTorch对应关系(参考官网)

    作者其他相关文章:Pytorch环境详细安装教程【Win10+CUDA升级11.6+cudNN+Anaconda3虚拟环境+pycharm】 官网: https://pytorch.org/get-started/previous-versions/ CUDA Toolkit版本及可用PyTorch对应关系总结(参考官网) CUDAToolkit版本 可用PyTorch版本 7.5 0.4.1 ,0.3.0, 0.2.0,0.1.12-0.1.6 8.0 1.1.0,1.0.0 ,0.4.1

    2024年02月06日
    浏览(63)
  • 关于GPU显卡、CUDA版本、python版本、pytorch版本对应

    提示:算力、cuda版本、python版本、pytorch版本有一项不匹配都无法使用 一、查看自己计算机的当前cuda版本以及gpu的算力 二、根据算力下载对应的cuda版本 三、根据算力以及CUDA版本下载查看需要的pytorch版本 四、下载可以兼容上述Pytorch版本的python 五、下载上述pytorch版本 查看

    2024年02月08日
    浏览(43)
  • 【stable-diffusion】4090显卡下dreambooth、lora、sd模型微调的GUI环境安装(cuda驱动、pytorch、xformer)

    相关博文 stable-diffusion-webui的基础功能手动安装,了解代码结构、依赖、模型出处 cuda11.7下载-https://developer.nvidia.com/cuda-toolkit-archive cudnn8.6 for 11.x : https://developer.nvidia.com/rdp/cudnn-archive dreamboothlora 训练环境: 最原始的命令端 https://github.com/kohya-ss/sd-scripts GUI端口 https://github.co

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

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

    2024年02月08日
    浏览(48)

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

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

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

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

二维码1

领取红包

二维码2

领红包