zoukankan      html  css  js  c++  java
  • Ubuntu16.04下安装tensorflow(GPU加速)【转】

    本文转载自:https://blog.csdn.net/qq_30520759/article/details/78947034

    版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq_30520759/article/details/78947034

    本文主要是通过tensorflow官网的教程来实现安装,如果无法安装的话也可以试着用源码去安装(源码安装相对比较复杂不推荐)。 
    1.前期的环境查询准备 
    1.1查看tensorflow的对于系统的基本要求 
    在安装任何一个软件的时候,我们首先应该去尝试了解它的运行环境,否则可能安装很久后到后面还是没有办法使用。 
    tensorflow的官方网址为:https://www.tensorflow.org/install/ 
    这里写图片描述
    我们看到tensorflow对于系统还是有一定的要求的,考虑到兼容性和开源资料的丰富度,我最后选择的系统的ubuntu16.04. 
    1.2选择在ubuntu系统上下载tensorflow 
    点击上图中的Installing Tensorflow on Ubuntu进入安装教程界面: 
    这里写图片描述
    考虑到计算和运行速度我这里选择Tensorflow with GPU support。 
    1.3NVIDIA requirements to run TensorFlow with GPU support 
    在ubuntu的安装教程界面有英伟达对于用GPU跑Tensorflow的一些要求。具体如下图几点,简单用中文翻译就是: 
    这里写图片描述

    • 1.CUDA®工具包8.0。 有关详细信息,请参阅NVIDIA的文档。 确保按照NVIDIA文档中所述将相关的Cuda路径名添加到LD_LIBRARY_PATH环境变量中。
    • 2.与CUDA Toolkit 8.0相关的NVIDIA驱动程序。
    • 3.CUDNN v6.0。 有关详细信息,请参阅NVIDIA的文档。 确保按照NVIDIA文档中所述创建CUDA_HOME环境变量。
    • 4.具有CUDA Compute Capability 3.0或更高版本的GPU卡。 请参阅NVIDIA文档以获取支持的GPU卡列表。

    在满足了以上信息后我们开始正式安装。 
    2.NVIDIA显卡驱动的安装 
    2.1打开终端先删除旧的驱动

    sudo apt-get purge nvidia*
    • 1

    2.2禁用自带的 nouveau nvidia驱动 (important!)

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

    并在文件中添加如下内容: 
    blacklist nouveau 
    options nouveau modeset=0 
    然后在更新一次 sudo update-initramfs -u 
    修改后需要重启系统。确认下Nouveau是已经被你干掉,使用命令: 
    lsmod | grep nouveau 结束X-window服务 
    首先我们需要结束x-window的服务,否则驱动将无法正常安装,通过以下命令实现。

    sudo service lightdm stop  
    • 1

    然后切换到tty1控制台:Ctrl+Alt+F1即可 
    2.4安装驱动

    cd /home/katherine/  
    sudo sh ./NVIDIA*.run 
    • 1
    • 2

    重新启动X-Window: sudo service lightdm start,然后Ctrl+Alt+F7进入图形界面 
    2.5检查

    nvidia-smi 
    • 1

    如果安装不成功,则需要卸载重来 sh ./NVIDIA.run --uninstall,或者可以尝试使用另一种方法。 
    在ubuntu系统中选择软件更新中的附加驱动来更新驱动。 
    这里写图片描述 
    按照上面图片点击即可,需要保证在联网的状态下,亲测可行。 
    3.gcc降版本 
    ubuntu的gcc编译器是5.4.0,然而cuda8.0不支持5.0以上的编译器,因此需要降级,把编译器版本降到4.9: 
    在terminal中执行:

    sudo apt-get install g++-4.9
    sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-4.9 20
    sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-5 10
    sudo update-alternatives --install /usr/bin/g++ g++/usr /bin/g++-4.9 20
    sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-5 10
    sudo update-alternatives --install /usr/bin/cc cc /usr/bin/gcc 30
    sudo update-alternatives --set cc /usr/bin/gcc
    sudo update-alternatives --install /usr/bin/c++ c++ /usr/bin/g++ 30
    sudo update-alternatives --set c++ /usr/bin/g++
    • 1
    • 2
    • 3
    • 4
    • 5
    • 6
    • 7
    • 8
    • 9

    4.安装CUDA 

    在cuda所在目录打开terminal依次输入以下指令(可以点进官网下载CUDA):

    cd  /home/***(自己的用户名)/Desktop/###(这个命令意思是找到刚刚我们用U盘传过来的文件)
    sudo dpkg -i cuda-repo-ubuntu1604-8-0-rc_8.0.27-1_amd64​.deb
    sudo apt-get update
    sudo apt-get install cuda
    • 1
    • 2
    • 3
    • 4

    5.安装CUDNN 
    打开terminal依次输入以下指令:

    cd  /home/***(自己的用户名)/Desktop/                     #(这个命令意思是找到刚刚我们用U盘传过来的文件)
    tar xvzf cudnn-8.0-linux-x64-v5.1-ga.tgz               #(解压这个文件)
    sudo cp cuda/include/cudnn.h /usr/local/cuda/include   # (复制)
    sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64     # (复制)
    sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
    • 1
    • 2
    • 3
    • 4
    • 5

    6.确定如何安装TensorFlow 
    这里写图片描述
    这里选择”native” pip,点击进去即可 
    6.1确定你电脑上python的版本 
    在安装ubuntu系统时,你的电脑会自动安装python,你必须确定你的电脑安装的是哪个版本:

    • Python 2.7
    • Python 3.4+ 
      pip或pip3软件包管理器通常安装在Ubuntu上。 花点时间确认(通过发出pip -V或pip3 -V命令)安装了pip或pip3。 我们强烈建议使用8.1或更高版本的pip或pip3。 如果未安装版本8.1或更高版本,请发出以下命令,该命令将安装或升级到最新的pip版本:
    sudo apt-get install python-pip python-dev   # for Python 2.7
    sudo apt-get install python3-pip python3-dev # for Python 3.n
    • 1
    • 2

    6.2开始下载tensorflow 
    6.2.1通过调用以下命令之一来安装TensorFlow

    pip install tensorflow            # Python 2.7; CPU support (no GPU support)
    pip3 install tensorflow          # Python 3.n; CPU support (no GPU support)
    pip install tensorflow-gpu    # Python 2.7;  GPU support
    pip3 install tensorflow-gpu  # Python 3.n; GPU support 
    • 1
    • 2
    • 3
    • 4

    如果上步失败,请通过发出以下格式的命令来安装最新版本的TensorFlow:

     sudo pip  install --upgrade tfBinaryURL   # Python 2.7
     sudo pip3 install --upgrade tfBinaryURL   # Python 3.n 
    • 1
    • 2

    7.完成安装开始验证 
    在terminal中输入:

    python
    • 1

    然后输入程序验证:

    import tensorflow as tf
    hello = tf.constant('Hello, TensorFlow!')
    sess = tf.Session()
    print(sess.run(hello))
    • 1
    • 2
    • 3
    • 4

    如果安装成功,则会显示 Hello, TensorFlow!

  • 相关阅读:
    动态规划——Best Time to Buy and Sell Stock IV
    动态规划——Split Array Largest Sum
    动态规划——Burst Ballons
    动态规划——Best Time to Buy and Sell Stock III
    动态规划——Edit Distance
    动态规划——Longest Valid Parentheses
    动态规划——Valid Permutations for DI Sequence
    构建之法阅读笔记05
    构建之法阅读笔记04
    构建之法阅读笔记03
  • 原文地址:https://www.cnblogs.com/zzb-Dream-90Time/p/9668045.html
Copyright © 2011-2022 走看看