PyTorch网站上说PyTorch 1.12.1与CUDA 11.6兼容，但我得到了以下错误：
NVIDIA GeForce RTX 3060 Laptop GPU 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.我使用笔记本电脑RTX 3060和诗歌作为我的Python包管理器。
第一个
我错过了什么？这是PyTorch〈〉CUDA问题还是CUDA〈〉GPU问题？
1条答案
按热度按时间sczxawaw1#
具有CUDA功能sm_86的NVIDIA GeForce RTX 3060笔记本电脑GPU与当前安装的PyTorch不兼容。当前安装的PyTorch支持CUDA功能sm_37 sm_50 sm_60 sm_70。
您安装的PyTorch版本不支持GPU的二进制。这是因为您使用的PyTorch的构建者选择了这样的构建方式。这不是CUDA版本或PyTorch版本的问题。只是许多框架构建时使用的二进制架构范围有限，以保持它们分发的包的大小较小。
NVIDIA提供了一种方法，通过在运行时JIT重新编译来支持向前兼容架构运行旧代码。不幸的是，标准的PyTorch构建系统并不使用它来保存构建发行版的空间，所以在这种情况下，这对您没有帮助。
您唯一的解决方案是为您的GPU提供适当的二进制支持。