机器学习踩坑之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

如果出现类似如下提示说明安装成功
在这里插入图片描述

猜你喜欢

转载自www.cnblogs.com/AWSG-Shaodw/p/12398630.html