查看是否安装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
重启。
https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html
https://developer.nvidia.com/cuda-downloads
这种方式不适合在“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
依次卸载
sudo dpkg -r libcudnn8-samples
sudo dpkg -r libcudnn8-dev
sudo dpkg -r libcudnn8
通过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
卸载
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