目录
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