Ubuntu+Cuda+cudnn安装

问题描述

实验室的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 11.7 Downloads

# 下载需要的版本
wget https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda_11.7.0_515.43.04_linux.run
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
wget https://developer.nvidia.com/downloads/compute/cudnn/secure/8.9.7/local_installers/11.x/cudnn-linux-x86_64-8.9.7.29_cuda11-archive.tar.xz

# 解压
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