【从零开始构建GPT模型】（二）Windows篇：CUDA安装教程

目录

一、确定计算机显卡可安装的CUDA版本

二、确定Pytorch2.0所需的CUDA版本

三、下载对应版本CUDA

四、安装CUDA

五、下载cuDNN

六、测试是否安装成功

---

注意：本文未使用conda安装，因为不够灵活。

一、确定计算机显卡可安装的CUDA版本

win键调出开始菜单，直接输入cmd，连续点击enter键两次，在命令行中输入

nvidia-smi.exe

如下图所示，我们需要的CUDA版本为，不高于12.4版本。

二、确定Pytorch2.0所需的CUDA版本

打开Pytorch2.0官方网站，Start Locally | PyTorch，找到如下目录，下拉页面。

可见如下官方版本要求表格

请根据自己的需求，安装对于的版本。

如我，Windows->Conda->Python->CUDA12.1

因为第一步确定了我需要的CUDA版本不高于12.4，所以，就直接遵循官方的12.1版本了。

在本专栏文章第一篇里，http://t.csdnimg.cn/fj6s1，已经介绍了Conda的安装了，需要的可以转到看一下。