如何在ubuntu 2210中安装NVIDIA驱动、CUDA和cuDNN(2204同样方法)

本篇讲解参考NVIDIA官方CUDA ToolKIT DOCUMENTATIONCUDNN DUCUMENTATION

本机参数如下:

Distributor ID:Ubuntu
Description:Ubuntu Kinetic Kudu (development branch)
Release:22.10
Codename:kinetic
Linux内核版本:5.15.0-27-generic
显卡型号:NVIDIA GeForce RTX3060 Mobile / Max-Q

安装NVIDIA显卡驱动

采用ubuntu自带的驱动安装,也可指定安装的驱动(建议指定驱动安装,不然可能会给你默认装server驱动)。

自动安装
# 自动安装
sudo ubuntu-drivers autoinstall
# 指定驱动安装
sudo apt install nvidia-driver-515

安装完后重启

sudo reboot

重启后输入以下命令查询显卡驱动是否安装成功,我们在这张图中可以看到当前驱动版本位515.48.07,CUDA版本最高可支持到11.7。由于我们的系统是最新的,所以我们默认全都装最高版本

nvidia-smi

在这里插入图片描述

手动安装

去Nvidia官网下载驱动:https://www.nvidia.com/en-us/drivers
在这里插入图片描述

进入命令行界面
sudo telinit 3
关闭显示服务
sudo service gdm3 stop
赋予权限并安装
sudo chmod a+x NVIDIA-Linux-x86_64-525.105.17.run
sudo ./NVIDIA-Linux-x86_64-XXX版本号.run -no-opengl-files -no-x-check -no-nouveau-check

可能出现的问题
安装过程中的选项:

问题:The distribution-provided pre-install script failed! Are you sure you want to continue?
选择 yes 继续。

问题:Would you like to register the kernel module souces with DKMS? This will allow DKMS to automatically build a new module, if you install a different kernel later?
选择 No 继续。

问题:Install Nvidia's 32-bit compatibility libraries?
选择 No 继续。

问题:Would you like to run the nvidia-xconfigutility to automatically update your x configuration so that the NVIDIA x driver will be used when you restart x? Any pre-existing x confile will be backed up.
选择 Yes 继续
打开显示服务
sudo service gdm3 start
重启
sudo reboot

安装CUDA 11.7

本机参数如下:

使用以下命令验证当前GPU的能力:

lspci | grep -i nvidia

使用以下命令检查当前Linux的版本支持:

uname -m && cat /etc/*release

查询内核支持:

sudo apt-get install linux-headers-$(uname -r)

以下安装均以初次安装,不提供卸载讲解

NVIDIA CUDA工具下载网站:https://developer.nvidia.com/cuda-downloads

在这里插入图片描述

使用以下命令删除过时的签名的密钥:

sudo apt-key del 7fa2af80

由于我们使用的是下载到本地安装方式,所以在 下载位置 输入以下指令:

sudo dpkg -i cuda-repo-你下载的那个文件(一般用Tab补齐就行).deb

安装完后会提示你需要注册临时公共秘钥,控制台中会给你一个命令,复制控制台上出现的!!

sudo cp /var/cuda-repo-(复制控制台上的命令)-local/cuda-*-keyring.gpg /usr/share/keyrings/

更新APT仓库缓存

sudo apt-get update

安装CUDA SDK

sudo apt-get install cuda

安装nvidia-gds

sudo apt-get install nvidia-gds

重启电脑

sudo reboot

开始设置环境变量

sudo gedit ~/.bashrc

在文件最后写入

export CUDA_DIR=/usr/local/cuda-11.7/bin${CUDA_DIR:+:${CUDA_DIR}}
export LD_CUDA_LIBRARY_PATH=/usr/local/cuda-11.7/lib64${LD_CUDA_LIBRARY_PATH:+:${LD_CUDA_LIBRARY_PATH}}

我们输入以下命令查询CUDA安装情况

nvcc -V

在这里插入图片描述

安装cuDNN 8.4

由于官方需要注册账号下载,下载地址:https://developer.nvidia.com/rdp/cudnn-download

在这里插入图片描述

下载完成后进入你的cuDNN路径中,解压CUDNN文件

tar -xvf cudnn-linux-x86_64-8.x.x.x(cuDNN版本)_cuda版本-archive.tar.xz

将以下文件复制到CUDA目录。

sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include 
sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64 
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐