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

问题描述:

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。

官网链接:

  • CUDA Toolkit Archive | NVIDIA Developer
  • cuDNN Archive | NVIDIA Developer

 具体安装方法参照win10 安装Pytorch GPU版+CUDA+cuDNN(篇幅较长保姆教程) (含CUDA11.1与cuDNN8.0.4安装包)_潮一的博客-CSDN博客_cuda11.1安装包

你可能感兴趣的:(pytorch,人工智能,python)