update-alternatives 下安装cudnn

检查是否安装显卡驱动

查看是否安装NVIDIA显卡

lspci | grep -i nvidia

查看显卡信息

nvidia-smi

驱动?

如果没有驱动,那么就需要手动安装显卡驱动了。

先添加源:

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

检查可安装的驱动:

ubuntu-drivers devices
sudo apt install nvidia-driver-XXX

安装系统推荐那个

sudo ubuntu-drivers autoinstall

重启。

安装CUDA

https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html
https://developer.nvidia.com/cuda-downloads

安装cudnn

方式1

这种方式不适合在“update-alternatives”下安装。

https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html#install-linux
https://developer.nvidia.com/rdp/cudnn-download

在这里插入图片描述

sudo dpkg -i cudnn-local-repo-ubuntu2004-8.9.2.26_1.0-1_amd64.deb

程序会被解包到var目录下

cd /var/cudnn-local-repo-ubuntu2004-8.9.2.26

在这里插入图片描述

sudo dpkg -i libcudnn8_8.9.2.26-1+cuda12.1_amd64.deb libcudnn8-dev_8.9.2.26-1+cuda12.1_amd64.deb libcudnn8-samples_8.9.2.26-1+cuda12.1_amd64.deb 

在这里插入图片描述

卸载deb文件

依次卸载

sudo dpkg -r libcudnn8-samples
sudo dpkg -r libcudnn8-dev
sudo dpkg -r libcudnn8

ubuntu上安装与卸载deb文件

通过deb包安装软件:

sudo dpkg -i package_file.deb

卸载:

sudo dpkg -r package_name

注意,卸载时候是package_file.deb对应的package name

若不知道package name,可以通过

dpkg -l查找,若要查找对应的package,可以加通配符,如查找包含fox的package

dpkg -l *fox*

但是这里对libcudnn8不合适,不知道为什么。

dpkg -l *libcudnn8
dpkg -l libcudnn8

找到之后,可以运行

dpkg -r package_name卸载

方式2

“update-alternatives”下安装。
在这里插入图片描述
解压:

tar -xvJf cudnn-linux-x86_64-8.9.3.28_cuda12-archive.tar.xz 

复制:

sudo cp cudnn-linux-x86_64-8.9.3.28_cuda12-archive/include/cudnn* /usr/local/cuda-12.2/include/
sudo cp cudnn-linux-x86_64-8.9.3.28_cuda12-archive/lib/libcudnn* /usr/local/cuda-12.2/lib64/

sudo chmod a+r /usr/local/cuda-12.2/include/cudnn*
sudo chmod a+r /usr/local/cuda-12.2/lib64/libcudnn*

查看一下版本信息:

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

猜你喜欢

转载自blog.csdn.net/philosophyatmath/article/details/131846117
今日推荐