深度学习(TensorFlow)环境搭建:(二)Ubuntu16.04+1080Ti显卡驱动

时间:2022-10-04 16:18:02

深度学习(TensorFlow)环境搭建:(二)Ubuntu16.04+1080Ti显卡驱动

  前几天把刚拿到了2台GPU机器组装好了,也写了篇硬件配置清单的文章——《深度学习(TensorFlow)环境搭建:(一)硬件选购和主机组装》。这两台也在安装Ubuntu 16.04和1080Ti显卡驱动,在安装Ubuntu的时候,踩过无数个坑,心力憔悴(...(。•ˇ‸ˇ•。)…),因此将踩过的坑以及对于的解决方案汇总出来,让大家少踩那些坑,过程实在是太磨人了。

一、配置

  • 系统:Ubuntu16.04.3
  • GPU:GTX1080Ti

二、总体流程步骤

  1.   安装Ubuntu16.04
  2.   安装1080Ti显卡驱动

三、安装Ubuntu16.04

  3.1  “曲线救国”-----低配N卡点亮系统

  以前安装Ubuntu也算是轻车熟路的,本以为这次也是一次性过,但是却踩了许多坑,这次使用UltraISO制作U盘启动盘,进行安装Ubuntu。但是刚开始就碰到了坑,已经在Blos设置里设置了U盘启动,但是怎么重启都看不到界面。一度怀疑是不是写入的U盘镜像有问题,后面在普通的PC机上试了下,是完全可以的。经过一番探索(直接问了主板卖家),原来配置的主板,第一;没有集成显卡,第二;Ubuntu 16.04安装是无法识别1080Ti的显卡的。如果你的主板也是这种情况的话,那我们只能“曲线救国”了,我们需要准备一块低配置的N卡插入主板,点亮系统,这样才能安装好系统。(如果没有低配想N卡,其实也可以将固态硬盘拆下了,放置在普通PC上进行安装系统,安装好之后再换回去),如图所示,是我拿的一块低配N卡先点亮系统。

深度学习(TensorFlow)环境搭建:(二)Ubuntu16.04+1080Ti显卡驱动

  将2块1080Ti的显卡换下了,用低配N卡换到主板上去,插上U盘启动安装系统(记得在BIOS设置U盘为第一启动项)如图所示:

深度学习(TensorFlow)环境搭建:(二)Ubuntu16.04+1080Ti显卡驱动

  不出意外,重启电脑都能看到系统安装界面,最体的如何安装Ubuntu这里就不做介绍了,网上也很多Ubuntu安装教程,大家可以参考下百度经验的一篇文章——《Ubuntu 16.04 安装基础入门教程》,推荐大家使用英文版,如果已经安装好了Ubuntu的话,大家千万别立马就把1080TI给换上去,否则你会发现重启电脑,界面一直停留在紫色的空白界面,如图所示:

深度学习(TensorFlow)环境搭建:(二)Ubuntu16.04+1080Ti显卡驱动

  这个时候系统处于假死的状态,那是因为,我们还需要在低配N卡环境下,把1080Ti的显卡驱动给安装到系统上才行。这就是接下来我们要将的如何安装1080TiGPU显卡驱动。


四、安装NVIDIA显卡驱动

4.1、查询NVIDIA版本号

  首先,我们可以去NVIDIA官网查看下,合适自己显卡驱动以及对应的版本号是多少。官网查询地址(http://www.nvidia.com/Download/index.aspx?lang=en-us),下图是我按照自己显卡类型查询的:

深度学习(TensorFlow)环境搭建:(二)Ubuntu16.04+1080Ti显卡驱动

  查询到1080Ti最新版为384.90(不过据其他网友反映最新版的384好像有问题,无法开机,这个我没有校验,所以保险起见,我还是安装了381的版本),如图所示为查询到的最新版本驱动:

深度学习(TensorFlow)环境搭建:(二)Ubuntu16.04+1080Ti显卡驱动

4.2、驱动安装

  Step 1:重启进入BIOS关闭secure boot,如果不关闭的话,驱动是不会生效的

  具体如果禁用BIOS的secure boot,根据自己实际情况设置,各个主板BIOS不一样,这里以MSI微星主板为例,重启电脑,按Delete键进入BIOS界面,具体secure boot是在【SETTINGS ==> Advanced ==> Windows OS Configuration ==> Secure Boot】,如图所示:

深度学习(TensorFlow)环境搭建:(二)Ubuntu16.04+1080Ti显卡驱动

  Step 2:禁用nouveau自带驱动

  注明:我的不需要禁用,所以这步其实我没有做,建议大家可以暂时跳过这步,如果后面发现安装了NVIDIA驱动还是不行,可以回到这步将nouveau禁用掉,禁用nouveau其实是因为Ubuntu系统集成的显卡驱动程序是nouveau,主要是怕和NVIDIA驱动有冲突,导致系统无法启动。

  这里我也讲解下如果禁用掉nouveau驱动

  控制台输入命令,创建一个文件通过命令

sudo vim /etc/modprobe.d/blacklist-nouveau.conf

  并添加如下内容:

blacklist nouveau

options nouveau modeset=0

  再更新一下

sudo update-initramfs -u

  修改之后需要重启系统。确认下nouveau是已经被你干掉,可以使用使用命令:

lsmod | grep nouveau

  Step 3:安装GPU驱动

  下文演示的还是在桌面模式(Ctrl+Alt+F7),真实环境下,实际上大家还是进入文本模式(Ctrl+Alt+F1),输需要输入账户和密码。

  1、安装ppa库

  可能有时候通过NVIDIA官网下载显卡驱动比较慢,我们可以添加ppa库,通过ppa安装显卡驱动,这步可有可无,用户自行决定,输入代码如下:

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

如图所示:

深度学习(TensorFlow)环境搭建:(二)Ubuntu16.04+1080Ti显卡驱动

  2、关闭图形化环境

  为了确保驱动能够正常安装,我们需要暂时关闭x-window服务(图形环境),在文本模式下输入命令进行关闭

sudo service lightdm stop

  3、安装驱动

  输入以下命令,直接安装驱动,我选的是之前说的381版本的

sudo apt-get update && sudo apt-get install nvidia-381

  如图所示:

深度学习(TensorFlow)环境搭建:(二)Ubuntu16.04+1080Ti显卡驱动

根据提示选择即可最后安装完毕后,重新启动X-Window,输入命令:sudo service lightdm start,然后Ctrl+Alt+F7进入图形界面。

备注:如果之前你已经安装NVIDIA显卡驱动话,我们需要先卸载旧的显卡驱动,需要输入命令:

sudo apt-get purge nvidia*

  至此、安装GPU显卡驱动已经完成,这个时候我们才可以把低配的N卡拿出来,换上我们的主角----1080Ti显卡,然后重启电脑,这个时候能看到已经顺利的进入系统。

  进入系统,我们要看刚刚的显卡驱动是否已经正确安装成功,通过下面命令查看:

nvidia-smi

  正常情况下,我们可以看到具体的驱动信息,如图所示:

深度学习(TensorFlow)环境搭建:(二)Ubuntu16.04+1080Ti显卡驱动

  至此、NVIDIA显卡驱动已经安装完成了~下篇我们主要讲CUDA和CUDNN的安装

PS:如有疑问,请留言,未经允许,不得私自转载,转载请注明出处:http://www.cnblogs.com/xuliangxing/p/7569946.html 

深度学习(TensorFlow)环境搭建:(二)Ubuntu16.04+1080Ti显卡驱动

深度学习(TensorFlow)环境搭建:(二)Ubuntu16.04+1080Ti显卡驱动