最近准备玩下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)
我的电脑的显卡型号为:GeForce RTX 2060 SUPER,知道显卡版本后开始安装显卡驱动。
1.2验证系统是否是受支持的Linux版本
uname -m && cat /etc/redhat-release
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选择自己适配的驱动
选择想要下载的驱动版本,并点击 “获取下载” 进行下载。
这些版本不同版本对应的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
重启后正确查看到驱动信息,自此说明安装成功了