问题描述:
GeForce RTX 3060 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.
问题原因:
没有或未正确安装CUDA和Cudnn。
解决办法:
根据自己的显卡,在英伟达官网下载适配的CUDA和Cudnn。
官网链接:
具体安装方法参照win10 安装Pytorch GPU版+CUDA+cuDNN(篇幅较长保姆教程) (含CUDA11.1与cuDNN8.0.4安装包)_潮一的博客-CSDN博客_cuda11.1安装包