如果你的电脑安装了 Ubuntu,而且电脑自带一块 NVIDIA GeForce 的 GPU 显卡,那么不用来跑深度学习模型就太可惜了!关于这方面的网上教程很多,但大都良莠不齐。这篇文章将手把手教你如何安装 GPU 显卡驱动值得一试!
注意:由于不同系统,不同 GPU 对应的 CUDA 和 cuDNN 均有差异,本文仅以 Ubuntu16.04、NVIDIA GeForce GPU为例,来介绍安装过程。其它版本过程类似,但存在细微差别,请读者以实践为主。
必要条件
工欲善其事必先利其器,首先确定你的电脑满足下面两个条件。
- 安装 Ubuntu16.04
- 电脑自带一块 NVIDIA GeForce GPU 显卡
一、安装 GPU 显卡驱动
有一块 NVIDIA GPU 显卡 还不够,还需要安装 GPU 显卡驱动。
1、首先登陆 GeForce 官网:
https://www.geforce.cn/drivers
在手动搜索驱动程序里,选择自己对于的系统,注意是“Linux 64-bit”。其他选项默认即可。然后,点击开始搜索,将会自动搜索你的电脑相匹配的 GPU 显卡驱动,搜索结果如下图所示:
如上图所示,结果会显示可用的 GPU 显卡驱动程序,确定你需要的版本,直接点击就可以下载了。例如我选择的版本是418.43,下载后的文件是 NVIDIA-Linux-x86_64-418.43.run。
2、打开终端 Terminal,首先卸载系统之前安装的 NVIDIA 显卡驱动,以防万一。(没有安装的可以忽略这一步)
代码语言:javascript
AI代码解释
$ sudo apt-get --purge remove nvidia-*3、打开 blacklist:
代码语言:javascript
AI代码解释
$ sudo gedit /etc/modprobe.d/blacklist.conf在最后添加几行:
代码语言:javascript
AI代码解释
blacklist vga16fb blacklist nouveau blacklist rivafb blacklist nvidiafb blacklist rivatv这时候在终端输入:
代码语言:javascript
AI代码解释
$ lsmod | grep nouveau如果没有输出,可以直接进行下一步,安装驱动。如果仍有输出,则在终端输入:
代码语言:javascript
AI代码解释
$ sudo gedit /etc/modprobe.d/blacklist-nouveau.conf直接写入:
代码语言:javascript
AI代码解释
blacklist nouveau blacklist lbm-nouveau options nouveau modeset=0 alias nouveau off alias lbm-nouveau off保存关闭,再在终端输入下面的命令验证是否没有输出了。
代码语言:javascript
AI代码解释
$ sudo update-initramfs -u $ lsmod | grep nouveau # 如果这样还有输出,reboot一次到这一步,应该没有输出了,可以进行安装驱动了。
4、在正式安装驱动之前,请记住你的驱动的 .run 文件的下载路径和完整的文件名。按 Ctrl+Alt+F1 进入命令行界面,输入下面的命令:
代码语言:javascript
AI代码解释
$ sudo /etc/init.d/lightdm stop $ sudo init 3进入 .run 文件的下载路径,使用 chmod 命令增加 .run 文件的运行权限。
代码语言:javascript
AI代码解释
$ chmod a+x NVIDIA-Linux-x86_64-418.43.run $ sudo ./NVIDIA-Linux-x86_64-418.43.run -no-x-check -no-nouveau-check -no-opengl-files #这里是关闭x server,nouveau和不下载opengl文件然后就会开始安装程序,安装完成之后,输入下面的命令:
代码语言:javascript
AI代码解释
$ nvidia-smi如果出现与下面类似的结果,证明驱动安装成功了。
最后,回到图像界面。
代码语言:javascript
AI代码解释
$ sudo /etc/init.d/lightdm start这样,驱动就算正式安装成功了。
原网址请见:https://cloud.tencent.com/developer/article/1933287