NVIDIA 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.

• 寫在最前面
• 項目場景：
• 問題描述
• 原因分析：
• 解決方案：
• • 查看gpu的算力（即nvidia的算力）
  • 查看pytorch版本&支持的cuda算力
  • 查看cuda版本
  • 查看對應(yīng)版本

The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70.)

寫在最前面

表面上是pytorch版本不夠，實際是pytorch所依賴的cuda版本不夠


總結(jié)：給RTX 3090配置cuda11以上版本即可

項目場景：

環(huán)境
RTX 3090 + linux
配置
torch-1.10.0 torchtext-0.11.0
cuda11.8

---

問題描述

UserWarning:
帶有CUDA能力sm_86的NVIDIA GeForce RTX 3090與當前的PyTorch安裝不兼容。
當前的PyTorch安裝支持CUDA功能sm_37 sm_50 sm_60 sm_70。
如果您想在PyTorch中使用NVIDIA GeForce RTX 3090 GPU，請查看https://pytorch.org/get-started/locally/的說明

UserWarning: NVIDIA 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 NVIDIA GeForce RTX 3090 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

---

原因分析：

參考：https://blog.csdn.net/Paramagnetism/article/details/115221478

幸福的煩惱
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版本下運行

---

解決方案：

參考：https://blog.csdn.net/weixin_41529093/article/details/122039547

查看gpu的算力（即nvidia的算力）

nvidia-smi nvidia-smi -a

nvidia官網(wǎng)的算力查詢
https://developer.nvidia.cn/zh-cn/cuda-gpus

查看pytorch版本&支持的cuda算力

>>> import torch >>> print(torch.__version__)

python import torch torch.cuda.get_arch_list()

[‘sm_37’, ‘sm_50’, ‘sm_60’, ‘sm_70’]

查看cuda版本

nvcc _V

查看對應(yīng)版本

pytorch官網(wǎng)
https://pytorch.org/get-started/previous-versions/

上pytorch官網(wǎng)查看（安裝）最新版本的cuda以及對應(yīng)的pytorch

# CUDA 11.3 conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=11.3 -c pytorch -c conda-forge

在現(xiàn)有conda下安裝報錯

因此新建環(huán)境

conda create -n wyt_1.10 python==3.8

進入到環(huán)境下

conda activate wyt_1.10

然后安裝，還是報錯
換成pip版本

pip install torch==1.10.0+cu111 torchvision==0.11.0+cu111 torchaudio==0.10.0 -f https://download.pytorch.org/whl/torch_stable.html

成功了

查看支持算力

總結(jié)

以上是生活随笔為你收集整理的幸福的烦恼：显卡算力太高而pytorch版本太低不支持的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。