ubuntu20.04安装CUDA、Cudnn

先检查显卡驱动是否安装

nvidia-smi

• 如上显示就代表驱动已安装完成，同时为下一步安装cuda确认信息，查看右上角CUDA Version的版本，我目前是12.8，就代表我目前可以安装的cuda版本不能高于12.8。
• 如果没有如上显示，就代表驱动未安装，可执行以下命令使用apt安装

添加ppa源

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update

查看可安装的驱动版本

ubuntu-drivers devices

建议直接使用recommended推荐版本

sudo apt install nvidia-driver-XXX

之后再次输入nvidia-smi，若还是看不到，可尝试重启

安装CUDA

CUDA Toolkit 12.9 Downloads | NVIDIA Developer

首先是确定版本，上述链接为nvidia安装最新cuda的网址。

按照之前说的，确认自己需要的版本，之前的版本可以从Archive of Previous CUDA Releases中找。

进入之后按照自己系统的配置选择选项

查看ubuntu版本可用以下命令

lsb_release -a

安装完成后还需要配置环境变量

sudo nano ~/.bashrc

在末尾添加如下两句,注意其中的xx需要更换为自己cuda的版本号

export PATH=/usr/local/cuda-xx.x/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-xx.x/lib64:$LD_LIBRARY_PATH

退出时使用ctrl+x,询问是否保存修改，输入y，再问写入文件名称，确认后直接回车

回到命令行，输入如下命令，让修改立即生效

source ~/.bashrc

最后输入下述命令，验证cuda的安装

nvcc -V

安装cuDNN

cuDNN 9.10.1 Downloads | NVIDIA Developer

安装cudnn类似cuda，直接进入官网按照命令下载即可，最后一步对照自己的cuda执行

如需验证，可先安装torch包，然后执行下述代码

import torch
print(torch.backends.cudnn.version())  # This will print the cuDNN version if available