windows环境下CUDA安装及配置

新电脑重新安装pytorch，记录从头配置的过程
参考博文：https://blog.csdn.net/weixin_43848614/article/details/117221384

安装对应版本的CUDA toolkit

查看对应版本

电脑打开NVIDIA控制面板

选择“帮助”中的“系统信息”

控制面板看版本号

通过“组件”查看对应安装cuda的版本，个人电脑是11.1

下载对应版本

cuda toolkit 链接
在列表中选择和自己电脑对应的cuda版本。进入下载页面

在下载页面中，确定相应配置，windows 10，选择“exe(local)"，进行下载

toolkit安装

双击exe进行安装，设置安装路径
进入安装页面，可以直接”精简模式“快速安装

这里我选择自定义为了设置安装路径

中间出现了因为visual studio 版本不匹配的问题，重新去下载了vs2019

配置好以后开始安装

安装完成

检查系统环境变量是否导入了cuda

命令行运行是否安装成功

nvcc --version
set cuda

完成！成功！

安装CudNN

下载对应版本

官网下载链接
进入NVIDIA主页，选择platforms，选择cudNN

选择download，需要先注册一下+填写问卷
都完成了以后就能进入下载页面了

如果没有符合自己cuda版本的cudnn，可以选择Archived releases 查看更多版本
这里我选择了下面的版本

安装

下载完成以后是一个压缩包，解压缩

将解压缩后的三个文件夹拷贝到cuda配置的路径中，就是set cuda 命令后展示的路径.两个文件夹中有同名的文件夹也没关系。直接复制即可。

添加系统环境路径

往系统环境变量中的 path 添加如下路径（根据自己的路径进行修改）
{cuda 安装路径} \bin
{cuda 安装路径} \include
{cuda 安装路径} \lib
{cuda 安装路径} \libnvvp

验证是否安装成功

首先win+R启动cmd，cd到安装目录下的 …\extras\demo_suite,然后分别执行bandwidthTest.exe和deviceQuery.exe


都得到pass，配置成功！