第一步 查看驱动版本
第二步 查看cuda版本
第三步 根据下图判断自己的CUDA版本是否符合要求(装的CUDA和cuDNN保持一致,版本号可以比驱动版本低)
第四步 进入官网https://developer.nvidia.com/rdp/cudnn-download ,先注册登录,查看自己的CUDA版本号对应自己的cuDNN
注意
通过清华源下载的pytorch是CPU版本,详情torch.cuda.is_available()返回false——解决办法_Nefu_lyh的博客-CSDN博客
终极大法
删环境,重装ptorch