前言

最近在玩 LLM
但是发现都没有安装 CUDA 驱动

CUDA 工具包

  • CUDA Toolkit 包含 NVIDIA 驱动
1
https://developer.nvidia.com/cuda-toolkit-archive

NVIDIA 驱动

  • 直接下载太旧,用 CUDA Toolkit 自带的 NVIDIA 驱动就可以
1
https://www.nvidia.cn/Download/index.aspx?lang=cn

cuDNN 驱动

1
https://developer.nvidia.com/rdp/cudnn-archive

安装方法

安装 CUDA 工具包

  1. 直接点击 .exe 文件下一步就行
  1. 同意并继续
  1. 直接选 精简 就OK
  1. 勾选,下一步
  1. 安装完毕

安装 cuDNN 驱动

  1. 查找 CUDA 文件路径
1
where nvcc
  1. cuDNN 文件解压后将文件放入 CUDA 驱动文件夹即可

查看 CUDA 安装效果

NVIDIA 控制面板

  • 右键 打开 NVIDIA 控制面板 --> 帮助 --> 系统信息 --> 组件

mvidia-smi

  • 参数不够详细
1
mvidia-smi

nvcc -V

1
nvcc -V