美文网首页
ubuntu安装TensorFlow

ubuntu安装TensorFlow

作者: Toplhyi | 来源:发表于2019-04-05 13:23 被阅读0次

    使用Python语言作为开发语言。

    1. 安装Anaconda
    bash Anaconda3-5.3.1-Linux-x86_64.sh
    
    • “Do you approve the license terms?”,输入yes。
    • 输入Anaconda的安装路径或者Enter选择默认的路径。
    • “Do you wish the installer to prepend the Anaconda3 install Location to PATH in your .../.bashrc?”,输入yes将anaconda3的binary路径加入到.bashrc文件。
    • 安装完成。在终端输入python命令,进入python shell。
    1. 安装Bazel
    • 安装JDK8。
    sudo add-apt-repository ppa:webupd8team/java
    sudo apt update
    sudo apt install oracle-java8-installer
    
    • 安装Bazel的其他依赖工具包。
    sudo apt install pkg-config zip g++ zlib1g-dev unzip
    
    chmod +x bazel-0.24.0-installer-linux-x86_64.sh
    ./bazel-0.24.0-installer-linux-x86_64.sh --user
    
    • 安装TensorFlow的其他依赖包。
    sudo apt install python3-numpy swig python3-dev python3-wheel
    

    Bazel安装完成后,会在home空间下生成名为bin的文件夹,里面是一个名为bazel的脚本文件。将bzael导入PATH:

    export PATH="$PATH:$HOME/bin"
    

    在终端输入bazel命令验证是否安装成功。
    ps:安装时出现了如下错误,原因是bazel安装包所在路径包含中文。

    Error: LinkageError occurred while loading main class com.google.devtools.build.lib.bazel.Bazel
        java.lang.ExceptionInInitializerError: null
    
    1. 安装CUDA。
      CUDA是NVIDIA推出的使用GPU资源进行复杂通用并行计算的SDK。
      可以在CUDA GPUs上查看自己显卡的计算能力值。
      显卡
      显卡计算能力值
    • NVIDIA官网获取CUDA Toolkit,页面还给出了安装方式。
      下载CUDA
      安装完成之后需要重启计算机使其生效。重启之后验证CUDA是否安装成功。首先在终端进入/usr/local/cuda/samples目录,执行以下make命令:
    make all
    

    如果make成功,会在最后提示“Finished building CUDA samples”信息。之后,在/usr/local/cuda/extras/demo_suite文件夹下,可以找到可执行文件deviceQuery并在终端运行该文件,会输出GPU相关信息。
    折腾了几天,并且重装了两次系统之后还是没有搞定,最终决定暂时放弃GPU。所以第4步跳过。

    1. 安装cuDNN。
    • NVIDIA官网获取cuDnn安装包。下载到的文件是cudnn-10.1-linux-x64-v7.5.0.56.tgz。
    • 运行tar -zxvf命令解压
    tar -zxvf cudnn-10.1-linux-x64-v7.5.0.56.tgz
    

    解压后会在当前目录生成一个名为cuda的文件夹,进入该文件夹。

    • 将文件夹include和lib64下的文件复制到CUDA的安装目录下相应的文件夹。
    sudo cp lib64/libudnn* /usr/local/cuda/lib64/
    sudo cp include/duDNN.h /usr/local/cuda/include/
    
    • 更新cuDNN库文件的软链接。
    cd /usr/local/cuda/lib64/
    sudo chmod +r libcuDNN.
    
    • 在系统环境里设置CUDA的路径。
      打开~/.bashrc文件,将如下代码插入到最下面:
    export   LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64:usr/local/cuda-10.1/extras/CPUTI/lin64:$LD_LIBRARY_PATH
    export CUDA_HOME=/usr/local/cuda-10.1
    export PATH=/usr/local/cuda-10.1/bin:$PATH
    

    之后执行source ~/.bashrc命令使之生效。

    1. 正式安装TensorFlow(使用pip安装)
    pip install --upgrade tensorflow
    
    1. 测试TensorFlow
      输入一下代码,输出结果:array([4,6], dtype=float32)
    import tensorflow as tf
    a = tf.constant([1.0,2.0],name='a')
    b = tf.constant([3.0,4.0],name='b')
    result = a + b
    sess = tf.Session()
    sess.run(result)
    

    如图:


    测试

    相关文章

      网友评论

          本文标题:ubuntu安装TensorFlow

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