Python 训练人脸识别模型(2) NVIDIA显卡驱动安装

来源:互联网 发布:问答系统 知乎 编辑:程序博客网 时间:2024/06/15 11:37

NVIDIA显卡驱动安装

为了使用深度学习的GPU版本。爬了好几次坑,终于装好了NVIDA的驱动。(在此,让我们一起跟着Linus竖起中指:Fuck you NVIDIA)。
博主的显卡是 GeForce GTX 950M,大家可以通过lspci | grep 3D 来查看各自的显卡。

其实有简单的安装方式:
系统设置 > 软件更新 > 附加驱动 > 选择nvidia 驱动进行安装。
不过,不推荐使用。因为博主安装之后就出现闪屏,开机黑屏的问题。
再一次竖起中指 —-> NVIDIA

下面就讲解一下如何可靠安装驱动:

1)打开终端,先删除旧的驱动:

sudo apt-get purge nvidia*

2)禁用自带的 nouveau nvidia驱动
修改一个文件通过
sudo vim /etc/modprobe.d/blacklist.conf

权限问题,就先 chmod 666 ,修改完内容之后再 chmod 644

并在文件最后添加如下

blacklist vga16fb
blacklist nouveau
blacklist rivafb
blacklist nvidiafb
blacklist rivatv

再更新一下 sudo update-initramfs -u
修改之后需要重启系统。确认下Nouveau是已经被你干掉,使用命令: lsmod | grep nouveau

添加官方ppa源

快捷键ctrl+alt+T打开命令终端,加入官方ppa源。

$ sudo add-apt-repository ppa:graphics-drivers/ppa

需要输入密码并按enter键确认。之后刷新软件库并安装最新驱动。

$ sudo apt-get update

接下来就是关键所在:
退出图形界面(Ctrl+Alt+F1),并关闭 服务 sudo service lightdm stop 。
接下来,由于之前已经更新过源,可以用 命令 ubuntu-drivers devices 查看支持的驱动列表。
然后用 sudo apt-get install nvidia-384 (博主的是这个,大家根据情况在列表中选择)
慢慢从网上下载、安装。如有问题,大多是网络问题,多次输入命令即可。(还有一种是从网上下载了驱动包,直接安装)
安装完成之后, reboot 一下。开机之后,开机之后就可以用 nvidia-settings 、 nvidia-smi 来查看是否安装正常。

(题外话)
双显卡更换使用,可以在nvidia-settings中的 PRIME Profiles 中进行选择,不过更换之后需要注销,重新登陆。当然也可以通过其他工具进行管理。