运行时报错如下图：

报错原因

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版本下运行

debug方法
直接上pytorch官网装最新版本的cuda以及对应的pytorch，肯定就可以让所有显卡都可以用。
下面展示一些 内联代码片。

pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html

安装11.1的cuda及对应pytorch即可解决问题
完成后conda list查看pytorch等版本：

若觉得有用请点个赞哦～谢谢～