机器学习踩坑之Ubuntu系统的Nvidia驱动安装

因为机器学习需要用到GPU进行运算，所以为Ubuntu系统安装Nvidia驱动就显得那么重要，在Ubuntu中安装Nvidia（下面统称为N卡）驱动过程中最恶心的无非是安装完以后出现Ubuntu系统循环登录问题，今天我就来跟大家分享一下我的踩坑经验。

声明

网上有很多简易方法比如什么输入一行代码

sudo apt-get install nvidia-*

这种方法虽然可以使用但是有很严重的问题，这样会造成登录界面循环，所以我采用这种麻烦的方法进行安装。

准备工作

• 我们需要确定自己电脑的N卡型号
  在终端中输入
  lspci | grep -i nvidia
  后面的中括号显示的就是Nvidia的显卡型号

• 按照N卡型号去Nvidia的官网下载驱动安装文件。

安装步骤

步骤如下

禁用nouveau驱动

安装N卡驱动之前需要先将nouveau驱动屏蔽才能安装N卡官方驱动，将驱动添加到黑名单blacklist.conf中就可以
由于这个文件的属性不允许修改，所以需要先修改文件属性

• 修改文件属性
  在终端中输入这行代码

sudo chomd 666 /etc/modprobe.d/blacklist.conf

• 用gedit编辑器打开blacklist.conf
  在终端中输入这行代码

sudo gedit /etc/modprobe.d/blacklist.conf

在该文件后添加一下这几行

blacklist vga16fb
 
blacklist nouveau
 
blacklist rivafb
 
blacklist rivatv
 
blacklist nvidiafb

保存文件

• 执行如下命令，更新系统，来禁用nouveau

sudo update-initramfs -u

• 重启电脑
• 重启成功后，在终端中输入如下命令检查是否禁用nouveau成功，如果没有输入，表示禁用成功。

lsmod | grep nouveau

按下ctrl+alt+f1进入命令行界面

• 在命令行界面输入如下指令，关闭图形界面（如果不执行后面会出错）：

sudo service lightdm stop

给驱动run文件赋予执行权限

sudo chmod a+x 文件名

安装

sudo ./filename -no-opengl-files

注意很多循环登录问题都是由于这个opengl导致的所以在这里直接将它去掉就可以避免，安装过程中还有许多选项
如果提示有旧驱动，询问是否删除旧驱动，选yes
如果提示缺少某某模块（modules），询问是否上网下载，选no
如果提示编译模块，询问是否进行编译，选ok
如果提示将要修改Xorg.conf，询问是否允许，选yes

挂载Nvidia驱动

modprobe nvidia

检查驱动是否安装成功

nvidia-smi

如果出现类似如下提示说明安装成功

参考文章：https://blog.youkuaiyun.com/wangyjfrecky/article/details/84029668