查看nvidia驱动版本
在windows终端键入nvidia-smi
,查看nvidia显卡驱动。显卡驱动的版本决定了CUDA的版本下限。
如果出现上述的问题,则终端进入C:\Program Files\NVIDIA Corporation\NVSMI文件夹内,再键入nvidia-smi
,可以看到我的驱动版本是472.84。
nvidia显卡驱动版本与CUDA的版本关系:
可以看到我当前的nvidia驱动版本,最高可以支持CUDA11.4.
更新nvidia驱动
如果说要安装高版本的CUDA,但驱动版本太旧,则需要更新Nvidia驱动。
需要进入Nvidia官网下载新的驱动:
https://www.nvidia.cn/geforce/drivers/
根据笔记本的显卡系列搜索驱动,如我的显卡版本是GeForce1060,当前为止,我的笔记本貌似最高可以安装到536.67。
下载下来之后,根据提示安装即可。安装完之后,通过查看nvidia-smi
驱动版本是否更新。
下载CUDA
进去cuda下载界面,下载所需的cuda版本。我这里选择的是CUDA11.4.
https://developer.nvidia.com/cuda-toolkit-archive
这里有本地安装和在线安装的区分。我选择本地安装,下载exe文件。
下载之后,双击即可开始安装;
等待一会,安装完之后,在conda的终端,键入nvcc -V
,验证是否安装成功。
下载cudnn
cudnn是cuda的深度学习高性能计算库,里面包含了深度学习常见算子的高性能实现。
https://developer.nvidia.com/rdp/cudnn-archive
解压cudnn文件夹,将里面的文件全都拷贝至cuda安装目录C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4。
拷贝的过程中,可能需要管理员权限。