【报错处理】NVIDIA GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch

 GeForce RTX 3090等高性能显卡配置pytorch时出现错误：

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_61 sm_70 sm_75 compute_37.

错误是当前3090显卡与torch的版本不支持。30系列的显卡不支持CUDA11.0以下的版本
解决办法就是更改torch版本。

PyTorch版本和对应的CUDA版本的关系在PyTorch官网上看。

假设使用的是RTX 3090，项目依赖的PyTorch为1.7.1

1. 查到对应的代码能力架构为sm_86 ，架构名为Ampere
2. 查到可以使用11.0-12.2版本的CUDA
3. 由下图，查到1.7.1版本的PyTorch对应的CUDA版本包括 CUDA-11.0、10.2等

4. 取交集，得到适合的CUDA版本为CUDA-11.0

5. 通过pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html安装正确版本的PyTorch库。

Pytorch 快速测试GPU状态

# 判断是否安装了cuda
import torch
from torch.backends import cudnn
 
print(torch.cuda.is_available())  # 返回True则说明已经安装了cuda
# 判断是否安装了cuDNN
print(cudnn.is_available())  # 返回True则说明已经安装了cuDNN
a = torch.cuda.is_available()
print(a)
ngpu = 1
# Decide which device we want to run on
device = torch.device("cuda:0" if (torch.cuda.is_available() and ngpu > 0) else "cpu")
print(device)
print(torch.cuda.get_device_name(0))
print(torch.rand(3, 3).cuda())
print('CUDA版本:', torch.version.cuda)
print('Pytorch版本:', torch.__version__)
print('显卡是否可用:', '可用' if (torch.cuda.is_available()) else '不可用')
print('显卡数量:', torch.cuda.device_count())
print('是否支持BF16数字格式:', '支持' if (torch.cuda.is_bf16_supported()) else '不支持')
print('当前显卡型号:', torch.cuda.get_device_name())
print('当前显卡的CUDA算力:', torch.cuda.get_device_capability())
print('当前显卡的总显存:', torch.cuda.get_device_properties(0).total_memory / 1024 / 1024 / 1024, 'GB')
print('是否支持TensorCore:', '支持' if (torch.cuda.get_device_properties(0).major >= 7) else '不支持')
print('当前显卡的显存使用率:', torch.cuda.memory_allocated(0) / torch.cuda.get_device_properties(0).total_memory * 100,'%')

粘贴复制后，直接运行，测试会得到以下类似果

True
True
True
cuda:0
NVIDIA GeForce RTX 3090
tensor([[0.0554, 0.2989, 0.6040],
        [0.3081, 0.9663, 0.6560],
        [0.3963, 0.8477, 0.4649]], device='cuda:0')
CUDA版本: 11.0
Pytorch版本: 1.7.1+cu110
显卡是否可用: 可用
显卡数量: 1
当前显卡型号: NVIDIA GeForce RTX 3090
当前显卡的CUDA算力: (8, 6)
当前显卡的总显存: 23.99951171875 GB
是否支持TensorCore: 支持
当前显卡的显存使用率: 0.0 %