【PyTorch】切记：GeForce RTX 3090 显卡仅支持 CUDA 11 以上的版本！

2024-06-05 06:22:42

问题描述

前不久给新来的 2台 8 张 GeForce RTX 3090 服务器配置了深度学习环境（配置教程参考这篇文章），最近在使用的时候却遇到了各种问题。

问题 1：GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch installation. The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70. If you want to use the GeForce RTX 3090 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

问题 2：RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

解决方案

原来是 GeForce RTX 3090 显卡仅支持 CUDA 11 以上的版本！而从官网得知 PyTorch 1.7.0 开始才支持 CUDA 11，所以要使用 GPU 训练的话，必须安装 PyTorch 1.7.0 及以上版本。

引用参考

CUDNN error: CUDNN_STATUS_EXECUTION_FAILED解决方法

【PyTorch】切记：GeForce RTX 3090 显卡仅支持 CUDA 11 以上的版本！相关推荐

零基础搭建Win系统Anaconda+Pytorch+OpenCV深度学习环境（Win10、Win11、RTX 3090显卡也适用）
需要工具: Anaconda.Pytorch.OpenCV和编程软件Pycharm. (跟着我们下面说的步骤配的都成功了,想想为什么别人能成功,而你就失败呢?) 1. 安装Anaconda: 点击进入 ...
玩家在RTX 3090显卡中发现了指套
点上方计算机视觉联盟获取更多干货仅作学术分享,不代表本公众号立场,侵权联系删除转载于:新智元 AI博士笔记系列推荐周志华<机器学习>手推笔记正式开源!可打印版本附pdf下载链接国外 ...
GeForce RTX 3090深度学习测评
GeForce RTX 3090深度学习测评环境踩坑八卡GeForce RTX 3090+Pytorch1.7+cuda11.1+对应cudnn pytorch 1.7以下版本无法对显卡写入数据 ...
AIStationV3.0 + GeForce RTX 3090 + 5280M5安装测试及故障处理
这篇记录的是AIStation安装过程中碰到的一些奇奇怪怪的报错之前做了个3090和服务器的适配测试,完事以后测试环境也没撤,正好最近有个大学AI实验室的实施项目要装AIStation(浪潮的人工智 ...
Ubuntu 18.04 安装 GeForce RTX 3090
Ubuntu 18.04 安装 GeForce RTX 3090 1,查看显卡型号 2,驱动安装 3,禁用nouveau 4,卸载显卡驱动重新安装 5,卸载nvidia cuda驱动安装驱动 ubu ...
暗黑破坏神3计算机性能测试,RTX 3090显卡8K游戏测试演示《暗黑3》稳定60帧运行...
RTX 3090显卡8K游戏测试演示 <暗黑3>稳定60帧运行 2020-11-01 17:29:00来源:游迅网编辑:血河评论(0) 对于主机玩家而言,今年次世代主机发售极大的增强了游玩 ...
PyTorch 1.7发布，支持CUDA 11、Windows分布式训练
机器之心报道参与:魔王.小舟昨日,PyTorch 团队发布 PyTorch 1.7 版本.该版本增添了很多新特性,如支持 CUDA 11.Windows 分布式训练.增加了支持快速傅里叶变换(FF ...
太棒了！PyTorch 1.7发布，支持CUDA 11、Windows分布式训练
点上方蓝字计算机视觉联盟获取更多干货在右上方 ··· 设为星标 ★,与你不见不散仅作学术分享,不代表本公众号立场,侵权联系删除转载于:机器之心 AI博士笔记系列推荐周志华<机器学习> ...
PyTorch 1.7发布：支持CUDA 11、Windows分布式训练
木易发自凹非寺量子位报道 | 公众号 QbitAI 自从7月份CUDA 11发布以来,就陆陆续续听到了网友类似的吐槽: 这正说着,10月27日,PyTorch团队发布了PyTorch 1.7 ...

最新文章

热门文章