zoukankan      html  css  js  c++  java
  • Ubuntu16.04+TensorFlow r1.12环境搭建指南

    一、操作系统安装

    OS版本:Ubuntu 16.04 (ubuntu-16.04.5-server-amd64.iso)

    CPU:4Core以上

    内存:4GB以上

    磁盘空间:80G以上

     

    二、基础环境准备

    1、为了提高apt安装速度,更新apt为国内阿里云源

    (1)复制原文件备份 
    sudo cp /etc/apt/source.list /etc/apt/source.list.bak

    (2)编辑源列表文件

    sudo vim /etc/apt/source.list

    (3)将原来的列表删除,添加如下内容

    deb http://mirrors.aliyun.com/ubuntu/ vivid main restricted universe multiverse 
    deb http://mirrors.aliyun.com/ubuntu/ vivid-security main restricted universe multiverse 
    deb http://mirrors.aliyun.com/ubuntu/ vivid-updates main restricted universe multiverse 
    deb http://mirrors.aliyun.com/ubuntu/ vivid-proposed main restricted universe multiverse 
    deb http://mirrors.aliyun.com/ubuntu/ vivid-backports main restricted universe multiverse 
    deb-src http://mirrors.aliyun.com/ubuntu/ vivid main restricted universe multiverse 
    deb-src http://mirrors.aliyun.com/ubuntu/ vivid-security main restricted universe multiverse 
    deb-src http://mirrors.aliyun.com/ubuntu/ vivid-updates main restricted universe multiverse 
    deb-src http://mirrors.aliyun.com/ubuntu/ vivid-proposed main restricted universe multiverse 
    deb-src http://mirrors.aliyun.com/ubuntu/ vivid-backports main restricted universe multiverse

    (4)运行sudo apt-get update

    2、为了提高pip安装速度,更新pip为国内阿里云源

    mkdir ~/.pip

    vi ~/.pip/pip.conf

    编辑pip.conf文件:

    [global] index-url = http://mirrors.aliyun.com/pypi/simple/

    [install] trusted-host = mirrors.aliyun.com

    3、安装gnome桌面(显示视频窗口需要)

    sudo apt install ubuntu-gnome-desktop

    4、安装git

    sudo apt install git

    5、加配SWAP到4GB以上(避免编译过程中分配内存失败,如果已经有超过4GB的swap空间,或者安装操作系统过程中已配置较大swap,可跳过这步)

    sudo dd if=/dev/zero of=/swap bs=4096 count=1M

    sudo mkswap /swap

    sudo swapon /swap

    sudo echo "/swap swap  swap  sw 0   0"  >> /etc/fstab

     

    三、安装编译工具Bazel

    1、安装依赖包

    sudo apt-get install pkg-config zip g++ zlib1g-dev unzip

    2、下载Bazel安装包

    wget https://github.com/bazelbuild/bazel/releases/download/0.18.1/bazel-0.18.1-installer-linux-x86_64.sh

    3、安装Bazel

    chmod +x bazel-0.18.1-installer-linux-x86_64.sh

    ./ bazel-0.18.1-installer-linux-x86_64.sh --user

    4、设置环境变量

    sudo vi ~/.bashrc

    在文件最后添加:

    export PATH=$PATH":~/bin"

    source ~/.bashrc

    四、安装TensorFlowflow

    1、安装virtualenv、virtualenvwrapper

    pip install --user virtualenv

    pip install --user virtualenvwrapper

    mkdir -p ~/Envs

    sudo vi ~/.bashrc

    在文件最后添加:

    export WORKON_HOME=~/Envs

    source /usr/local/bin/virtualenvwrapper.sh

    source ~/.bashrc

    创建虚拟环境:mkvirtualenv tfenv

    进入虚拟环境:workon tfenv

    退出虚拟环境:deactivate

    2、安装依赖包:

    (1)准备Python依赖

    pip install -U six numpy wheel mock portpicker scipy futures scikit-learn grpcio grpcio-tools enum34

    pip install -U keras_applications==1.0.5 --no-deps

    pip install -U keras_preprocessing==1.0.3 --no-deps

    pip install -U --upgrade setuptools

    easy_install distribute

    pip install -U --upgrade distribute

    修正虚拟环境中的pkg_resources包

    workon tfenv

    cdsitepackages

    vi pkg_resources/__init__.py

    找到:from . import py31compat

    替换为:from pkg_resources import py31compat

    rm -f pkg_resources/__init__.pyc

    (2)准备go环境

    移除老版本:sudo apt remove golang-go

    安装go1.10

    sudo add-apt-repository ppa:gophers/archive

    sudo apt-get update

    sudo apt-get install golang-1.10-go

    设置go环境变量

    sudo vi ~/.bashrc

    在文件最后添加:

    找到export PATH=$PATH":~/bin"

    替换为:export PATH="$PATH:~/bin:/usr/lib/go-1.10/bin"

    source ~/.bashrc

    3、准备TensorFlow源码

    git clone https://github.com/tensorflow/tensorflow.git

    cd tensorflow

    git checkout r1.12  # r1.9, r1.10, etc.

    4、编译前测试

    bazel test -c opt -- //tensorflow/... -//tensorflow/compiler/... -//tensorflow/contrib/lite/...

    如果告警提示支持avx、avx2、fma、sse4.1、sse4.2,则使用如下命令:

    bazel test -c opt --copt=-mavx --copt=-mavx2 --copt=-mfma  --copt=-msse4.1 --copt=-msse4.2 --copt=-mfpmath=both  -- //tensorflow/... -//tensorflow/compiler/... -//tensorflow/contrib/lite/...

    5、编译配置

    在tensorflow源码根目录运行:

    ./configure

    配置如下:

     

    6、Build the pip package

    bazel build --config=opt //tensorflow/tools/pip_package:build_pip_package

    7、Build the package

    ./bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg

    8、Install the package

    pip install /tmp/tensorflow_pkg/tensorflow-version-tags.whl

     

    参考:

    https://www.tensorflow.org/install/source?hl=zh-cn

    https://docs.bazel.build/versions/master/install-ubuntu.html

    https://github.com/golang/go/wiki/Ubuntu

     

  • 相关阅读:
    详解package-lock.json的作用
    Cisco计网实验配置总结
    使用Vue制作了一个计算机网络中子网划分部分的简陋计算工具
    Prettier-Code Formater代码格式化插件使用教程
    Node.js中npx命令的使用方法、场景
    从几道题目带你深入理解Event Loop_宏队列_微队列
    简单模拟实现javascript中的call、apply、bind方法
    git使用说明书
    使用闭包模拟实现AMD模块化规范
    if执行后else if即使满足条件也不再执行
  • 原文地址:https://www.cnblogs.com/dskit/p/9939312.html
Copyright © 2011-2022 走看看