【Bug解决】NVIDIA GeForce RTX 3090 with CUDA capability sm_86 is not compatible with ...

报错描述

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/

在这里插入图片描述

错误描述: CUDA capability sm_86 是说当前 GPU 3090 算力是 8.6,与安装的 PyTorch 版本不匹配。虽然错误描述里说的是 PyTorch,但实际上是说 PyTorch 依赖的 CUDA 版本(因为我们安装的是 CUDA 版本的 PyTorch,如果使用 CPU 版的就不会有这个问题)。所以说,当前的 PyTorch CUDA 版本支持算力为只有 3.7、5.0、6.0、7.0,低于 GPU 3090 算力。

关于 GPU 算力和 CUDA 算力的关系是:

  • 算力 8.0 的显卡可以在最高支持算力 8.6 的 CUDA 下正常运行
  • 但算力 8.6 的显卡不能在最高支持算力 8.0 的 CUDA 下正常运行
  • 也就是说 CUDA 支持的最高算力要大于等于GPU本身的算力

关于版本问题:

  • CUDA 10.x 最高支持算力为 7.x,CUDA 11.x 最高支持算力为 8.x

报错环境:

  • Linux:Ubuntu 18.04
  • GPU:NVIDIA GeForce RTX 3090
  • CUDA:11.3
  • Pytorch:1.9.0
  • Python:3.8.10
  • GCC:7.5.0

解决方案

在 PyTorch官网 下载合适的 Pytorch 版本,CUDA 11.6 最高支持算力是 8.6,所以就下载支持 CUDA 11.6 以上的 PyTorch 版本(目前最高也就是 11.6 了)。
【Bug解决】NVIDIA GeForce RTX 3090 with CUDA capability sm_86 is not compatible with ..._第1张图片

# 首先卸载已有的pytorch
pip uninstall torch
pip uninstall torchvision
pip uninstall torchaudio
# 安装
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116

安装成功,查看 torch 版本:
在这里插入图片描述

类似的问题还有:【Bug解决】nvcc fatal : Unsupported gpu architecture ‘compute_86‘

你可能感兴趣的:(#,PyTorch,pytorch)