1、CUDA和驱动的对应版本
查看地址:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
由于ChatGLM 采用的深度学习框架是Pytorch2.3,本次安装是GPU版本,对应的CUD为12.1
2、显卡信息
1 打开“任务管理器“”,性能下面查看显卡的硬件信息
2 查看驱动版本,如果版本不符合请更新显驱动
方式1: 显卡控制面板
方式2:快捷指令
终端中输入:nvidia-smi
3、安装显卡驱动
如果显卡版本达到要求可以略过本条
下载地址:https://www.nvidia.cn/Download/index.aspx?lang=cn&ptid=1&psid=52&pid=0