【Ubuntu显卡驱动及cuda更新】

一、 显卡驱动安装包

1.1 一般都有驱动，如果驱动版本过低，可以利用cuda安装包里的驱动安装。
1.2 不用卸载原有驱动,直接连着驱动和cuda一起安装。
cuda下载地址，选择runfile(local)
1.3 放在全英文路径下，方便后续查找，提升安装包权限，例如：

sudo chmod a+x cuda_12.0.0_525.60.13_linux.run

1.4 关闭图像化界面。

sudo service lightdm stop

1.5 进入tty模式。

ctrl + Alt +F1

1.6 安装，若需要升级显卡驱动，选择如下两个，安装过程可能提示已经有cuda 选择替换即可。
Driver
CUDA Toolkit

1.7 进入图像化界面。

sudo service lightdm start

1.8 删除安装包创建的cuda 软链接，可能会有问题，重新建立软链接。

ln -s /usr/local/cuda-12.0 /usr/local/cuda

二、 安装pytorch

2.1 根据cuda版本，下载对应的pytorch。
pytorch官网

三、 安装cudnn

3.1 pytorch对cudnn版本有要求，所以查看安装的pytorch所需要的cudnn版本。

import torch
print(torch.version)
print(torch.version.cuda)
print(torch.backends.cudnn.version())

3.2 下载对应的cudnn。
cudnn安装地址

3.3 安装cudnn。
先解压，后复制过去，并修改权限。
注意~/.bashrc,对应的环境变量指向的是软链接cuda。
但cudnn替换的是实际的cuda安装位置，例如 /usr/local/cuda-12.0

sudo cp cuda/include/cudnn.h /usr/local/cuda-12.0/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-12.0/lib64/
sudo chmod a+r /usr/local/cuda-12.0/include/cudnn.h
sudo chmod a+r /usr/local/cuda-12.0/lib64/libcudnn*