嵌入式深度学习框架之Tensorflow(二)编译&安装

2019-07-12 18:24发布

参考文档

  1. https://tensorflow.google.cn/install/install_sources Installing TensorFlow from Sources
  2. https://tensorflow.google.cn/install/install_linux Installing TensorFlow on Ubuntu

Tensorflow训练环境安装

训练环境概述:

  • CPU:Intel i7-4790K
  • GPU:1 x Nvidia GTX1060 & 1 x Nvidia GTX1080ti
  • 操作系统:Ubuntu 16.04.3 64位
  • GPU驱动版本:384.98
  • CUDA版本:9.0.176
  • cuDNN版本:7.0.4.31
  • 环境搭建时间:2018-04-12

源代码编译

下载源代码 git clone https://github.com/tensorflow/tensorflow cd tensorlfow git checkout r1.7
可以通过git的checkout选项切换到指定的Release
或者直接从github上下载稳定版本,并解压得到源代码。

安装Bazel

参考Bazel官方安装文档的建议,推荐使用apt方式安装Bazel。

安装jdk8

对于Ubuntu16.04: sudo apt-get install openjdk-8-jdk 对于Ubuntu14.04则需要先添加PPA源再进行安装: sudo add-apt-repository ppa:webupd8team/java sudo apt-get update && sudo apt-get install oracle-java8-installer

添加Bazel安装源

sudo apt-get install curl echo "deb [arch=amd64] http://storage.googleapis.com/bazel-apt stable jdk1.8" | sudo tee /etc/apt/sources.list.d/bazel.list curl https://bazel.build/bazel-release.pub.gpg | sudo apt-key add -

安装Bazel

sudo apt-get update sudo apt-get install bazel 如果安装过程中出现连接googleapi网站失败的问题,可以尝试多试几次。

升级Bazel

建议安装完Bazel之后进行版本升级 sudo apt-get upgrade bazel

安装Python依赖

sudo apt-get install python-numpy python-dev python-pip python-wheel

配置Tensorflow

cd tensorflow-1.7.0 ./configure

编译pip包

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

手动创建pip包

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

安装pip包

sudo pip install /tmp/tensorflow_pkg/tensorflow-1.7.0-cp27-cp27mu-linux_x86_64.whl

验证安装过程是否正确

进入python命令行执行环境 python 执行以下python代码,如果执行正确则表示tensorflow已经正确安装: import tensorflow as tf hello = tf.constant('Hello, TensorFlow!') sess = tf.Session() print(sess.run(hello))