报错描述
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 了）。
# 首先卸载已有的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‘