Ubuntu22.4+Cuda11.7+cuDNN8.9.7安装

问题描述

实验室的Ubuntu主机重启之后显示器分辨率显示异常，双屏只亮了一个，另外一个显示无信号，估计是显卡驱动的问题。
打开nvidia-smi显示找不到驱动，遂从网上搜集重新安装驱动以及cudnn的教程。此文记录此过程，方便后续查阅。

软硬件版本：Rtx3090 + Ubuntu22.04

1. 安装驱动

# remove nivida*
sudo apt-get --purge remove nvidia*

# install nvidia-driver
# 安装指定版本：
sudo apt install nvidia-driver-525
# 安装推荐版本：
sudo ubuntu-drivers autoinstall

2. 安装cuda

CUDA Toolkit Downloads，选择你的版本这里以11.7为例

# 下载需要的版本
wget https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda_11.7.0_515.43.04_linux.run

# 安装cuda，按照指引进行，注意Driver可以不安装。
sudo sh cuda_11.7.0_515.43.04_linux.run

# 设置环境变量
sudo gedit ~/.bashrc

# 添加
export PATH="/usr/local/cuda-11.7/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-11.7/lib64:$LD_LIBRARY_PATH"

# 激活
source ~/.bashrc

# reboot
rebbot

# then
nvidia-smi

3. 安装cudnn

cuDNN Archive Downloads

# 下载cuDNN v8.9.7 for CUDA 11.x，随后解压文件，需要登陆验证nvidia账户。
tar -xvf cudnn-linux-x86_64-8.9.7.29_cuda11-archive.tar.xz

# 将cudnn.h等文件复制到 /usr/local/cuda中
cd cudnn-linux-x86_64-8.9.7.29_cuda11-archive
sudo cp include/* /usr/local/cuda/include/
sudo cp lib/* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

# 测试安装是否成功
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

参考文章：
https://zhuanlan.zhihu.com/p/711028014
https://blog.csdn.net/sinat_16423171/article/details/140351150