windows安装cuda11.1、cudnn及pytorch
cuda驱动程序的下载与安装
查看显卡驱动版本是否符合cuda11.1的要求
cuda11.1要求显卡驱动版本>=455.23,如果不想升级驱动版本可安装低版本的cuda。如果当前版本不符合要求,可在待会cuda安装中勾选安装显卡驱动。
可在nvidia控制面板-系统信息-显示-细节中查看驱动程序版本。
下载cuda驱动安装包
点击此处跳转cuda下载链接
点击CUDA Toolkit 11.1.1,跳转到新页面,选择相应的信息即可开始下载。
安装cuda
运行安装包,打开需要等一会才会弹出界面,然后按下面图片所示选择安装:
- 选择临时解压路径,这个默认就行
- 选择同意并继续
- 选择自定义
- 当前驱动版本这也会显示当前版本的,安装包提供的版本是456.81,还没我462.30高,所以我就取消了勾选。如果驱动版本低于最开始那个表格中要求的版本就务必勾选上,满足了的话勾与不勾就随意啦。
- 选择安装路径,这个根据自己的需求改,我自己一般就选择默认的了。
- 打勾,点击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中下载即可。