美文网首页linux运维
Ubuntu 18.04 Server 搭建tensorflow

Ubuntu 18.04 Server 搭建tensorflow

作者: 吟风者 | 来源:发表于2019-09-16 10:04 被阅读0次

    一、系统安装

    参考ubuntu-server系统安装

    二、显卡驱动安装

    显卡型号:Nvidia GTX1050.

    1、首先去Nvidia官网下载驱动:

    image

    找到自己的显卡,选择Linux版本进行下载,下载的是.run文件。

    2、禁用nouveau

    安装NVIDIA需要把系统自带的驱动禁用,打开文件:

    sudo gedit /etc/modprobe.d/blacklist.conf
    

    在文本最后添加以下内容:

    blacklist nouveau
    option nouveau modeset=0
    

    命令窗口会提示warn,无视之。

    保存退出,执行以下命令生效:

    sudo update-initramfs -u
    

    重启电脑后输入:

    lsmod | grep nouveau
    

    没有任何输出说明禁用成功。

    3、安装NVIDIA显卡驱动

    在安装驱动程序的过程中,会因为缺少gcc、g++、make等development tool而报错导致无法完成驱动程序安装(ERROR:Ubable to find the development tool 'make' in your path...),不用担心,手动安装这些开发包后再次执行安装指令即可。另外,安装过程中出现的弹框根据默认选项选择即可。手动安装development tool指令为:

    sudo apt-get install gcc
    sudo apt-get install g++
    sudo apt-get install make
    

    4、 查看显卡驱动版本

    可以通过nvidia-smi命令查看GPU和驱动程序信息,若出现以下结果,则表明驱动程序安装成功。

    安装时没有截图,所以用了网上的图片

    三、安装docker

    1、安装基础docker服务

    安装基础的docker请参考Ubuntu下通过docker部署springboot项目 安装和配置docker部分。

    2、安装nvidia-docker2服务

    想要在docker容器中使用宿主机的gpu,则需要安装一个nvidia-docker的插件。下面介绍安装方法:
    第一种:

    wget -P /tmp https://github.com/NVIDIA/nvidia-docker/releases/download/v1.0.1/nvidia-docker_1.0.1-1_amd64.deb
    sudo dpkg -i /tmp/nvidia-docker*.deb
    

    第二种:

    sudo apt-get install nvidia-docker2
    

    本人通过第二种方式安装,第一种方式未经测试。
    可以直接去nvidia-docker 的github看看:https://github.com/NVIDIA/nvidia-docker/

    3、测试nvidia-docker

    nvidia-docker run --rm nvidia/cuda nvidia-smi
    

    然后在执行下面这句,默认用nvdia-docker替代docker命令:

    echo 'alias docker=nvidia-docker' >> ~/.bashrc
    bash
    

    最后就可以下载tensorflow镜像进行测试了,这里就不详细介绍了。
    附赠 tensorflow docker hub 地址

    相关文章

      网友评论

        本文标题:Ubuntu 18.04 Server 搭建tensorflow

        本文链接:https://www.haomeiwen.com/subject/zrtsyctx.html