最近准备玩下AI 图片生成，入手了二手2060s显卡。

1.准备
1.1查看显卡信息

lspci | grep -i nvidia
#03:00.0 VGA compatible controller: NVIDIA Corporation TU106 [GeForce RTX 2060 SUPER] (rev a1)
#03:00.1 Audio device: NVIDIA Corporation TU106 High Definition Audio Controller (rev a1)
#03:00.2 USB controller: NVIDIA Corporation TU106 USB 3.1 Host Controller (rev a1)
#03:00.3 Serial bus controller [0c80]: NVIDIA Corporation TU106 USB Type-C UCSI Controller (rev a1)

image.png

我的电脑的显卡型号为：GeForce RTX 2060 SUPER，知道显卡版本后开始安装显卡驱动。

1.2验证系统是否是受支持的Linux版本

uname -m && cat /etc/redhat-release

image.png

1.3查看系统是否有GCC编译环境

gcc -v

如果没有任何输出，需要安装gcc

yum install gcc

2.禁用Nouveau驱动

lsmod | grep nouveau

如果有任何输出，那么就是nouveau在启用，需要关闭，按照以下步骤：

#打开如下文件
sudo vim /usr/lib/modprobe.d/dist-blacklist.conf
#写入以下内容
blacklist nouveau
options nouveau modeset=0
#保存并退出
:wq
#重启
reboot
#最后输入上面的命令验证
lsmod | grep nouveau

没有任何输出，说明nouveau已禁用了
3.安装驱动
3.1官网下载驱动：
驱动
3.2选择自己适配的驱动

image.png

image.png

这些版本不同版本对应的cuda版本不一样，对应的cudnn版本也就不一样，支持的pytorch版本也会不一样，所以，目前选择哪个版本都可以，但是之后安装cuda、cudnn、pytorch的时候需要进行版本对应。
3.3安装驱动
进入下载的目录，运行下载的驱动程序，我下的是NVIDIA-Linux-x86_64-525.116.03.run。

sh NVIDIA-Linux-x86_64-525.116.03.run

如出现报错,关于 x 的报错。

chmod +x NVIDIA-Linux-x86_64-525.116.03.run
# -no-x-check  #安装驱动时关闭X服务
sh NVIDIA-Linux-x86_64-525.116.03.run -no-x-check

-安装过程有一些选项，基本上有默认的选默认的，有些查一下看看意思选，里面有四个选择需要去做
he distribution-provided pre-install script failed!Are you sure you want to continue?
选择“ok”
-Would you like to register the kernel module sources with DKMS?This will allow DKMS to auomatically 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-xconfig utility to automatically update your X configuration file so that the NVIDIA X driver will be user whten you rest…”
选择 No。

3.4查看驱动信息

reboot
nvidia-smi

image.png

重启后正确查看到驱动信息，自此说明安装成功了