zoukankan      html  css  js  c++  java
  • ubuntu16.04深度学习环境的配置【转】

    本文转载自:https://my.oschina.net/u/3837179/blog/1920756

    在ubuntu中配置GPU的深度学习环境相较于win问题要多很多,这几天琢磨了一下Ubuntu下的环境配置,参考很多人的博客,也遇到了不少坑,好不容易配置成功了,希望写下来,和大家分享,避免大家走弯路。环境的配置主要是nvidia显卡驱动的安装,在驱动安装的过程中遇到了问题,可以参考博客后面的问题解决方案,主要步骤就是装nvidia驱动,然后是安装cuda和cudnn,这两个一般问题不大。配置成功后就可以根据需要安装自己使用的深度学习框架(比如目前比较火的tensorflow和pytorch)。这两个框架本人建议使用Anconda安装,一来方便安装,二来如果出错,可以直接删除anconda文件夹重新安装即可。

    软件要求

    Ubuntu16.04
    Nvidia gtx 750 Ti
    Cuda 9.0
    Cudnn 7.0

    Nvidia显卡驱动的安装

    1 禁用显卡驱动nouveau

    网上很多文章说在官网下载run文件安装,这样安装的成功率较低,不建议这样做。还有在出现循环登录,黑屏时好多博客说卸载重装,这样也不大提倡,具体解决这些问题看博客后面的解决方案,亲测有效。由于nvidia的显卡驱动是闭源软件,因此ubuntu上默认是没有的,自带的显卡驱动叫做nouveau,安装之前要将nouveau先禁用掉,在终端里输入:

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

    在vim编辑器中输入以下两行:

    blacklist nouveau
    options nouveau modset=0

    然后配置文件生效,终端中输入:

    sudo update-initramfs –u

    2. 接下来安装一些后面会用到的32位lib,终端中依次输入

    sudo apt-get install lib32ncurses5
    sudo apt-get install lib32z1 

    然后重启,输入:

    sudo reboot

    3. 再次进入系统

    同时按下Ctrl+Alt+F1进入tty模式,登录账号,输入用户名密码。然后关闭图形界面服务,输入:

    sudo service lightdm stop

    接下来添加nvidia驱动的ppa源并进行安装:

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

    查看可安装的驱动版本,选择合适的版本安装,这里安装的384版本,

    ubuntu-drivers devices

    安装nvidia驱动及其依赖的包,

    sudo apt-get update sudo apt-get install nvidia-384
    sudo apt-get install mesa-common-dev
    sudo apt-get install freeglut3-dev

    打开图形界面:

    sudo service lightdm start

    禁用Secure boot的问题(非常非常重要!!!好多博客说在BIOS中禁用Secure boot,这样也可以,但会影响到win的启动,所以直接使用mok工具即可,如果这一步操作不成功,就九九归一了,滑稽):

    sudo moktuil --disable-validation

    然后输入8-16位密码(这里最好用简单的密码,本人之前设置很复杂的密码,结果在后面的mok manager界面输入密码时经常出错,所以如:88888888这样的密码很容易成功) 然后重启,输入:

    sudo reboot

    如果出现蓝色界面,选择改变secure boot状态的选项,进去提示输入对应位的密码(如:enter 2 characters:则输入密码的前两位),输入几遍后最后选择yes启动。

    打开终端,输入,nvidia-smi和nvidia-settings如果输出如下图,说明安装成功:

    这时nvidia就安装好了,如果碰到黑屏,循环登录,参考后面的问题解决方案

    Cuda的安装

    找到下载文件的路径

    sudo sh cuda_9.0.176_384.81_linux.run

    回车,一路往下运行,直到提示“是否为NVIDIA安装驱动nvidia-384?”,选择否,因为已经安装好驱动程序了,其他的全都是默认,安装位置默认是/usr/local/cuda,配置环境变量,运行如下命令打开profile文件

    sudo sh cuda_9.0.176_384.81_linux.run

    打开文件后在文件末尾添加路径,也就是安装目录,命令如下:

    export  PATH=/usr/local/cuda-9.0/bin:$PATH
    export  LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64$LD_LIBRARY_PATH

    保存,然后重启电脑

    sudo reboot

    测试CUDA的例子

    cd  /usr/local/cuda-9.0/samples/1_Utilities/deviceQuery
    sudo make
    ./deviceQuery

    如果显示的是关于GPU的信息,则说明安装成功了。 最后你会看到cuda驱动、sample、tookit已经安装成功,但是缺少一些库。

    环境变量配置 安装完毕后,再声明一下环境变量,并将其写入到 ~/.bashrc 的尾部:

    export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
    exportLD_LIBRARY_PATH=/usr/local/cuda9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

    然后设置环境变量和动态链接库,在命令行输入:

    sudo gedit /etc/profile

    在打开的文件末尾加入:

    export PATH=/usr/local/cuda/bin:$PATH

    保存之后,创建链接文件:

    sudo gedit /etc/ld.so.conf.d/cuda.conf

    在打开的文件中添加如下语句:

    /usr/local/cuda/lib64

    然后执行

    sudo ldconfig

    使链接立即生效

    cudnn的安装

    1) 继续Ctrl+alt+F1进入的字符界面 2) 进入解压后的cudnn-9.0-linux-x64-v7.0.tgz文件cuda,在终端执行下面的指令安装:

    tar -zxvf cudnn-9.0-linux-x64-v7.0.tgz
    cd cuda
    sudo cp lib64/lib* /usr/local/cuda/lib64/
    sudo cp include/cudnn.h /usr/local/cuda/include/

    然后更新网络连接:

    cd /usr/local/cuda/lib64/
    sudo chmod +r libcudnn.so.7.0.3  # 自己查看.so的版本
    sudo ln -sf libcudnn.so.7.0.3. libcudnn.so.7
    sudo ln -sf libcudnn.so.7 libcudnn.so
    sudo ldconfig

    重新启动图形化界面

    sudo service lightdm start

    再Ctrl+alt+F7退出Text Mode

    问题解决方案

    在安装nvidia显卡驱动的过程中会遇到各种各样的问题,这里主要说明一下常见的3个问题:

    问题1: 安装Ubuntu的时候,卡在Ubuntu的LOGO界面或黑屏 

    问题2: 双显卡安装Nvidia驱动,循环登录或黑屏 

    问题3: 启动后黑屏,并出现ubuntu the system is running in low-graphics mode 的错误

    针对以上3个问题,我们这里提出了相应的解决方案。

    注意!!!:对于问题1和问题2,如果下面对应的方法无效,则去掉”quiet splash”后面的apciosi或者nomodeset,在vthandoff后面添加acpi_osi或者nomodeset!!!

    方案1(针对问题1):

    1. 启动系统后在Grub界面,选择ubuntu系统的那一行,然后按E键,就会进入Grub的编辑状态。

    2、在quiet splash 后面加(先打空格)nomodeset,然后按F10保存启动

    3、由于那个是临时改动,所以还需要一步,打开终端:

    sudo vi /etc/default/grub

    编辑打开的文件,找到GRUBCMDLINELINUX_DEFAULT那一行,在后面加上(在quiet splash后打一个空格) nomodeset,保存,然后在终端输入 :

    sudo update-grub

    重启即可。

    方案2(针对问题2):

    问题的根源是集显的问题(但由于我们是双系统或双显卡或着存在Grub启动项,导致没有解决到问题的根本!) 这一切的原因就是因为我们装系统的时候,加的那个参数nomodeset 接下来详细教程 1、 首先确定是否nvidia驱动已安装,按Ctrl+Alt+F1进入tty模式,登录,然后输入nvidia-smi如果输出显卡信息,说明安装成功了。 2、重新启动,在Grub界面选Ubuntu系统那一行然后按E键进入编辑模式:

    3、也是上图那个位置,在 “quiet splash nomodeset”,主要看是否有nomodeset,有的话删除它(可能有人是queit splash= nomodeset,删除’= nomodeset’即可!) 4、然后在原来那个位置加 acpi_osi=linux(代码之间用空格隔开!),然后按F10启动,就可以进入界面了。

    5、最后一步,保存信息,打开终端:

    sudo vi /etc/default/grub 或 sudo gedit /etc/default/grub
    #打开文件后将nomodest删除替换为 acpi_osi=linux
    #然后更新grub
    sudo update-grub

    解释一下 Grub引导了系统进行启动,所以它的参数被传入了,即nomodeset(调用集显)如果存在,系统就会一直调用集显,然后就出现循环登录或黑屏。由于刚刚安装系统一般没有驱动,很多人只能通过调用集显去进入图形界面(除非在命令行下安装了驱动),导致了nomodeset参数的加入。 而acpi_osi=linux是告诉Grub,电脑将以Linux系统启动,调用其中驱动,所以可以用Nvidia的驱动进行显示了!

    方案3(针对问题3):

    当开机出现ubuntu the system is running in low-graphics mode错误时,按下Ctrl+Alt+F1,登录

    输入:

    cd /etc/X11

    然后输入:

    sudo cp xorg.conf.failsafe xorg.conf

    重启:

    sudo reboot
  • 相关阅读:
    什么是DMI,SMBIOS,符合SMBIOS规范的计算机的系统信息获取方法
    Android init.rc执行顺序
    JVM-类的四种载入方式
    Intellij-创建Maven项目速度慢
    Intellij-工程目录下隐藏不想显示的文件和文件夹
    JVM-类加载机制(Java类的生命周期)
    Git-远程仓库的使用
    JavaSE-反射-获取类或者对象的四种方法
    工厂模式(Factory Pattern)
    Redis-配置认证密码
  • 原文地址:https://www.cnblogs.com/zzb-Dream-90Time/p/9671747.html
Copyright © 2011-2022 走看看