说在前面:
NVIDIA的显卡驱动与CUDA的版本并不是严格的一一对应关系,CUDA实际上也只是一个工具包,我们可以根据自己的需求进行安装,即可以安装多个CUDA版本。同时CUDNN是一个SDK,专门用于神经网络的加速包,它与CUDA也没有严格的一一对应关系,可以选择多个版本的CUDNN进行安装。虽然说不是严格的对应,不过还是要服从大版本的对应,即
安装之前查看GPU驱动版本与CUDA版本的对应关系,选择对应的版本进行安装,避免不兼容而报错的错误。
查看版本对应关系https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
查看服务器显卡驱动版本:
nvidia-smi
查看显卡信息:
lspci | grep -i nvidia
下载CUDA: