目录

1. 查看显卡型号

2. 查看显卡算力

3. 查看算力对应的CUDA Runtime的版本范围

4. 安装显卡驱动最新版本

5. 查看CUDA Driver Version

6. 确定要安装的pytorch的cuda版本

---

1. 查看显卡型号

任务管理器 -> 性能 -> GPU ，右上角就是自己的显卡型号

2. 查看显卡算力

网址：https://en.wikipedia.org/wiki/CUDA

需要外网才能进去

里面有个表格，截图如下，找到自己显卡型号对应的算力，第一列是算力，第二列是显卡的架构

我的显卡算力是7.5，架构是Turing

3. 查看算力对应的CUDA Runtime的版本范围

下表是算力对应的CUDA Runtime的版本范围

从表中能够看出，7.5的算力对应的CUDA Runtime的版本范围是10.0-12.5

4. 安装显卡驱动最新版本

英伟达官网：https://www.nvidia.cn/

进入官网，点击右上角的驱动程序

在搜索栏里填入自己的显卡型号，然后点击查找

我直接选择了GeForce Game Ready驱动程序，点击查看

然后点击下载

双击打开下载好的安装包

选择驱动保存位置，点击OK

5. 查看CUDA Driver Version

打开命令行，输入下列指令，（可以是Windows键+R键，输入cmd，回车，打开的Windows命令行，也可以是Anaconda Prompt命令行），我用的Anaconda Prompt命令行，右上角的CUDA Version就是CUDA Driver Version

nvidia-smi

下图，是我安装显卡驱动最新版本之前的CUDA Driver Version，是11.6

下图是安装完显卡驱动最新版本后查看的CUDA Driver Version是12.5

6. 确定要安装的pytorch的cuda版本

显卡是硬件，不能直接被计算机识别，需要安装对应的驱动，才能被计算机识别。安装显卡驱动的时候就会安装CUDA Driver。CUDA是可以让显卡进行并行运算的软件。pytorch中的cuda是CUDA Runtime。pytorch中的CUDA Runtime去调用CUDA Driver，CUDA Driver去调用显卡。就是有CUDA Driver和显卡，计算机才能用显卡，有CUDA Runtime和CUDA Driver和显卡，pytorch才能用显卡。所以，

CUDA Runtime Version 要<= CUDA Driver Version

pytorch现在高版本的cuda有12.1，11.8，我选择cuda12.1