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

GeForce RTX 3080 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. If
you want to use the GeForce RTX 3080 GPU with PyTorch, please check
the instructions at
https://pytorch.org/get-started/locally/

意思是cuda版本算力和pytorch版本算力不符，cuda支持sm_86,pytorch最高sm_75

我先是检查了cuda版本，用命令

nvcc -V

发现我的cuda是11.1 没问题
所以我点开报错提示的链接https://pytorch.org/get-started/locally/重新下载对应的pytorch
命令如下

conda install pytorch torchversion torchaudio cudatoolkit=11.1 -c pytorch