1.确认有 Nvidia GPU
首先,你需要确认你的电脑是否安装了支持 CUDA 的 Nvidia GPU。可以通过以下步骤检查:
-
打开命令行(Windows 的
cmd
或 PowerShell,Linux 的终端)。 -
输入以下命令:
-
nvidia-smi
如果系统中安装了 Nvidia 驱动,并且 GPU 支持 CUDA,你会看到类似以下的输出:
-
在右上角红圈位置会显示你可以安装的CUDA版本,安装版本小于这个版本号即可,但一定是大版本一致,例如你的CUDA Version为12.XXX,你安装的CUDA版本也要为12.XXX
-
2.安装 CUDA
根据上图的CUDA版本进入CUDA官网CUDA Toolkit Archive | NVIDIA Developer下载符合自己电脑的CUDA,这个链接里可以看到历史版本,可根据需要下载