GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch installation.

最新推荐文章于 2024-05-11 15:24:16 发布

Steven山椒鱼

最新推荐文章于 2024-05-11 15:24:16 发布

阅读量3w

点赞数 65

文章标签： pytorch cuda

本文链接：https://blog.csdn.net/Paramagnetism/article/details/115221478

版权

运行时报错如下图：
报sm_86不匹配
报错原因

CUDA capability sm_86：算力8.6
上面表面上是说PyTorch，实际上是PyTorch依赖的CUDA版本的问题
翻译一下就是：RTX 3090的算力是8.6，但是当前的PyTorch依赖的CUDA版本支持的算力只有3.7、5.0、6.0、7.0

算力7.0的显卡可以在支持最高算力7.5的CUDA版本下运行，但是算力7.5的显卡不可以在支持最高算力7.0的CUDA版本下运行
同理算力8.x的显卡不可以在支持最高算力7.x的CUDA版本下运行

debug方法
直接上pytorch官网装最新版本的cuda以及对应的pytorch，肯定就可以让所有显卡都可以用。
在这里插入图片描述下面展示一些 内联代码片。

pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html

安装11.1的cuda及对应pytorch即可解决问题
完成后conda list查看pytorch等版本：
在这里插入图片描述
若觉得有用请点个赞哦～谢谢～

关注