ubuntu使用最佳流程2：ubuntu20.04安装cuda（多版本切换），cudnn，显卡驱动

cuda安装（多版本cuda安装：可切换）

查看系统硬件配置

查询Linux系统的版本号

lsb_release -a

查询显卡型号

待更新

下载

CUDA官方传送门

找到适合自己的命令行下载安装即可

安装

accept
第一个driver去掉（点击enter）

配置环境变量

写入环境变量

(1) 打开配置文件

sudo gedit ~/.bashrc

(2) 写入所需的环境变量

# CUDA enviroment variable
export PATH=$PATH:/usr/local/cuda/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda

(3) 更新环境变量

source ~/.bashrc

软连接

(1) 进入 /usr/local 目录下

cd /usr/local

(2) 删除原有的软连接

sudo rm -rf cuda

(3) 新建软连接，将其连接到 CUDA11.8（你想要切换的那款）

sudo ln -s /usr/local/cuda-11.8 /usr/local/cuda

验证安装

nvcc -V

安装 cudnn

1.下载

传送门
找到适合自己版本的

2.解压

(1) 进入下载目录

(2) 解压 .xz

xz -d cudnn-linux-x86_64-8.9.7.29_cuda11-archive.tar.xz

(3) 解压 .tar

tar -xvf cudnn-linux-x86_64-8.9.7.29_cuda11-archive.tar

3.替换文件

(1) 复制 lib 或 lib64 文件夹下的文件到 CUDA 安装位置的 lib64 文件夹下

sudo cp lib/* /usr/local/cuda-12.0/lib64/

(2) 复制 include 文件夹下的全部文件到 CUDA 安装位置的 include 文件夹下

sudo cp include/* /usr/local/cuda-12.0/include/

4.验证安装

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

出现下图提示即为安装成功

驱动安装

法1：查看推荐安装驱动版本

ubuntu-drivers devices

sudo apt-get install nvidia-driver-535
# 535 为驱动的版本号，可以在 ubuntu-drivers devices 中查看推荐版本

点击回车

reboot # 重启

法2：手动下载安装

下载

传送门

下载后复制到特定文件夹

mkdir ~/download
mv NVIDIA-Linux-x86_64-430.26.run ~/download

准备工作

安装必要软件

sudo apt update
sudo apt install build-essential

卸载原有驱动

sudo apt-get remove --purge nvidia*

禁用nouveau

sudo apt install vim
sudo vim /etc/modprobe.d/blacklist.conf

在blacklist末尾添加

blacklist nouveau
options nouveau modeset=0

输入如下更新

sudo update-initramfs -u

然后重启reboot

重启后输入

lsmod | grep nouveau

没有输出即禁用成功

安装

查看配置

确定当前系统使用的display manager：

$ cat /etc/X11/default-display-manager
# /usr/sbin/gdm3 此为输出

比如本系统使用的就是gdm3

安装显卡驱动

进入非图形化界面

sudo telinit 3

关闭显示

sudo service gdm3 stop

接着安装

cd ~/download
sudo chmod a+x NVIDIA-Linux-x86_64-550.78.run
sudo ./NVIDIA-Linux-x86_64-550.78.run --no-opengl-files 

安装过程中

1.The distribution-provided pre-install script failed! Are you sure you want to continue?
选择continue installation

2.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 继续。

3.问题大概是：Nvidia’s 32-bit compatibility libraries?
选择 No 继续。

4.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.
选择 No 继续

安装完成，开启显示

sudo service gdm3 start

打开终端，输入nvidia-smi即可看到显卡驱动安装完成

禁用系统和内核更新

由于ubuntu默认启动了自动更新内核，会使得我们的显卡驱动版本无法匹配内核版本，导致经常需要重装显卡驱动

sudo vi /etc/apt/apt.conf.d/10periodic
sudo vi /etc/apt/apt.conf.d/20auto-upgrades

将所有参数值都设为0，然后重启系统

禁用系统更新

打开Software & Updates（软件和更新）, 按照如下设置