Nvidia3090显卡驱动+CUDA+cuDNN安装步骤

显卡驱动

sudo add-apt-repository ppa:graphics-drivers/ppa   # 添加Nidia镜像
sudo apt update

ubuntu-drivers devices   ## 检查驱动适配型号

== 省略…
vendor : NVIDIA Corporation
driver : nvidia-driver-495 - distro non-free
driver : nvidia-driver-460 - distro non-free
driver : nvidia-driver-470-server - distro non-free
driver : nvidia-driver-460-server - distro non-free
driver : nvidia-driver-470 - distro non-free recommended
driver : xserver-xorg-video-nouveau - distro free builtin

sudo ubuntu-drivers autoinstall  ## 自动安装

sudo apt install nvidia-470

CUDA

进入cuda官网，选择所需的版本号
输入命令安装，并接受协议：

向bashrc中添加环境变量：

export PATH=/usr/local/cuda/bin/:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.4/lib64/

使生效：

source ~/.bashrc

查看cuda版本：

nvcc -V

CUDA安装结束,测试安装是否成功：

cd /usr/local/cuda/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery

安装成功～

cuDNN

进入cuDNN官网，需要注册并登陆

解压，将目录下两个文件夹下的内容复制到相应目录

sudo cp cuda/lib64/* /usr/local/cuda/lib64
sudo cp cuda/include/* /usr/local/cuda/include

查看cuDNN的版本信息

cat /usr/local/cuda/include/cudnn_version.h

完成安装