新手保姆级GPU版本pytorch安装（win系统），CUDA和torch对应版本！

提示：文章写完后，目录可以自动生成，如何生成可参考右边的帮助文档

目录

前言

一、CUDA安装

二、cuDNN安装

三、配置环境变量

---

前言

有一天跑yolo系列，出现cuda无效的bug，然后更新了环境，发现训练速度很慢，后来发现是自己搞错了环境，弄成CPU版本的了。所以就综合一下网上的教程和自己踩得坑，写一篇pytorch（GPU版本）详细教程，也是希望自己之后弄崩环境后能有个参考吧！

---

一、CUDA安装

1.CUDA驱动查看

电脑桌面右键，打开NVIDIA控制面板

 点击左下角系统信息

 点击组件，查看nvidia cuda驱动版本

 可以看到我的是11.6，目前来说11.6已经够用了，但是如果你的版本相对较低，建议更新一下驱动版本。

2.CUDA驱动更新

首先进入nvidia官方驱动的官网：官方驱动 | NVIDIA

然后点击搜索

看一下这里的型号是不是和自己电脑的GPU一样的，不一样就换成和自己GPU一样的型号即可。然后点击搜索后出现下面的下载界面，点击下载。

将下载好的驱动点击安装，安装步骤就是默认下一步（安装地址可以根据自己电脑实际情况来，默认c盘）。

更新好驱动后，继续查看驱动版本。同时按住win+R键，输入cmd进入终端，输入nvidia-smi，查看对应版本。 可以看到此时我的驱动版本对应的是512.15,cuda版本对应的11.6。

3.CUDA安装

（1）查看安装版本

首先进入CUDA Toolkit 官方界面：CUDA 12.1 Release Notes

向下划找到以下表格。

我的驱动版本是512.15，除了前两个cuda12以上的不能安装，其他都可，但是前面cuda版本是11.6，所以要安装的cuda版本不能超过11.6！这里作者选择cuda11.6版本进行安装。（下载的CUDA版本一定要是在11.6以下。）

（2）CUDA下载

CUDA官网下载地址：CUDA Toolkit 12.1 Downloads | NVIDIA Developer

进来之后点击查看历史版本

进来之后找到自己想安装的CUDA版本，本文选的是cuda11.6.0

点击相应的版本后，win10就选10，win11系统就选11，选择相应版本如下图即可下载

（3）CUDA安装

点击下载好的.exe文件，系统检查点击下一步，接下来安装选项，选择自定义，然后点击下一步。

选择安装地址的时候，可以选择默认的，也可以安装在其他地址，但一定要记住这个地址，因为后面配置环境的时候要用到这个地址！！！

后面就进行下一步直到安装结束就可以了。

 查看是否安装成功，同时按住win+R键，输入cmd，进入终端，输入nvcc -V,如果输出如下，即安装成功。

二、cuDNN安装

进入到cuDNN官网：https://developer.nvidia.com/rdp/cudnn-download

点击Archive of previous Releases。如果上面链接点击后没出现这个界面，就要注册一个账号，然后再登录就到这个界面啦。

 

接着点击cuDNN8.x-1.x。

查找CUDA11.6对应cuDNN版本，只要是11.x的都可以，我这里选的是8.5.0。其他版本也可以

下载，解压，之后将cuDNN文件中这三个文件夹中的内容分别复制到cuda的对应文件夹中

cuda的安装地址在前面安装的时候是有的，可以截图。或者默认。就是前面这个地方的安装地址

三、配置环境变量

首先在设置里面搜索环境变量，打开编辑系统环境变量。

 打开环境变量。

找到系统变量，path，点击

 

将下面的cuda安装路径新建到环境变量中，注意，这里是根据自己的安装路径填。

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\include
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\lib
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\lib\x64

添加后，点击确定。即可完成。