windows安装cuda11.1、cudnn及pytorch

cuda驱动程序的下载与安装

查看显卡驱动版本是否符合cuda11.1的要求

cuda11.1要求显卡驱动版本>=455.23，如果不想升级驱动版本可安装低版本的cuda。如果当前版本不符合要求，可在待会cuda安装中勾选安装显卡驱动。

可在nvidia控制面板-系统信息-显示-细节中查看驱动程序版本。

下载cuda驱动安装包

点击此处跳转cuda下载链接

点击CUDA Toolkit 11.1.1，跳转到新页面，选择相应的信息即可开始下载。

安装cuda

运行安装包，打开需要等一会才会弹出界面，然后按下面图片所示选择安装：

1. 选择临时解压路径，这个默认就行

2. 选择同意并继续
   
3. 选择自定义
   
4. 当前驱动版本这也会显示当前版本的，安装包提供的版本是456.81，还没我462.30高，所以我就取消了勾选。如果驱动版本低于最开始那个表格中要求的版本就务必勾选上，满足了的话勾与不勾就随意啦。
   
5. 选择安装路径，这个根据自己的需求改，我自己一般就选择默认的了。
   
6. 打勾，点击next，然后等待安装完成。
   

cudnn的下载与安装

下载cudnn

点此跳转到cudnn下载页面，需要登陆登录nvidia账号才行，没有可以注册一个。之后选择点击第一个就行了，如果想下载其他版本的cudnn，点击下方Archived cuDNN Releases即可。

选择对应的版本开始下载，这里选择如图红色方框所标示的版本就行了。

安装cudnn

cudnn下载下来后是一个压缩包，如果cuda是安装在默认路径，则将压缩包中bin文件夹的内容复制到下面的路径。如果是非默认路径，复制到到相应安装路径下的bin文件夹中就行了。

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\bin

将将压缩包中include文件夹的内容复制到到下面的路径。如果是非默认路径，复制到到相应安装路径下的include文件夹中就行了。

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\include

将将压缩包中lib文件夹的内容复制到到下面的路径。如果是非默认路径，复制到到相应安装路径下的对应的文件夹中就行了。

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\lib\x64

pytorch的下载与安装

点此进入到pytorch主页，在此页面选择系统，cuda版本等信息会显示安装的方式。可使用conda和pip安装，个人喜好我一般使用pip，如果使用conda建议更换国内源（推荐交大源或者清华源），这样下载更快。

在cmd中进入相应的环境，输入
pip3 install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio===0.8.1 -f https://download.pytorch.org/whl/torch_stable.html即可安装。

如果只想安装CPU版本，可在Compute Platform这选择CPU，根据相应命令在cmd中下载即可。