Windows下安装Pytorch和CUDA及配置CUDNN教程

一、查看电脑的显卡驱动版本

1.键盘点击Windows+R或者在搜索栏输入cmd，进入命令提示符，输入nvidia-smi，查看显卡驱动版本

如图所示，本电脑显卡驱动版本为:536.99

二、查找合适的CUDA版本以及cuDNN版本

（一）、查找合适的CUDA版本

1.点击此链接https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html，进入CUDA Toolkit官方文档界面，向下翻阅，找到以下图表：

可以看到本电脑最高可以装12.2版本的CUDA，保险起见，我选择安装11.6版本的

（二）、查找合适的cuDNN版本

1.点击此链接https://developer.nvidia.com/rdp/cudnn-download，进入cuDNN Archive界面下，查找适合CUDA11.6的cuDNN版本

我们cuda是11.6，这里就选择cuDNNV8.9.3版本的for CUDA11.x版本即可三、下载CUDA和cuDNN