tensorflow离线源码安装

从源码离线编译安装tensorflow,记录踩过的坑:1.tensorflow的外部依赖tensorflow的外部依赖很多,所依赖的外部库均在WORKSPACE文件和//tensorflow/workspace.bzl文件中给出了url地址使用如下命令可以拉取全部的external外部依赖bazel fetch //...http://blog.csdn.net/huang_yx005/artic

从源码离线编译安装tensorflow,记录踩过的坑:

1.tensorflow的外部依赖

tensorflow的外部依赖很多,所依赖的外部库均在WORKSPACE文件和//tensorflow/workspace.bzl文件中给出了url地址

使用如下命令可以拉取全部的external外部依赖

bazel fetch //...

http://blog.csdn.net/huang_yx005/article/details/50914735

在外部联网机器中使用脚本下载全部的外部依赖,并拷贝到内部局域网的http服务的根目录下。

将WORKSPACE文件和//tensorflow/workspace.bzl文件中的外部依赖的url地址的前缀全部换成内部局域网的http服务的地址。

至此,编译安装需要外部依赖时会从内部http下载相应的库


2.换行符问题

从github上clone下来的文件的可能采用dos换行符,也可能采用unix换行符

如果某个可执行文件采用了dos换行符,此时在liunx下运行就会出错

解决方法:

使用vim打开,输入vim命令 :set ff=unix

上述命令将文件中所有的dos换行符'\r\n'转换成unix换行符‘\n’


3.运行如下的命令时产生错误

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

error in tensorflow setup: 'install_requires' must be a string or list of string containing valid project/version requirement specifiers;

Expected ',' or end-of-list in backposts.weakref  >= 1.0rc1;

解决办法:

python版本高于3.4,python安装backposts.weakref.10rc1


4.源码编译的大致流程:

export TEST_TMPDIR=...

export LD_LIBRARY_PATH=...   需包含cuda, cudnn相关的路径

./configure   对tensorflow进行配置

bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package  编译目标程序,开启GPU

 If you would like to prevent new dependencies from being added during builds, you can specify the --fetch=false flag

上述bazel命令建立了一个名为build_pip_package的脚本。

运行如下的命令将会在 /tmp/tensorflow_pkg路径下生成一个.whl文件:

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

使用如下命令安装生成的pip包。

pip install /tmp/tensorflow_pkg/tensorflow-1.1.0-py2-none-any.whl

验证。打开任意一个新的终端,注意不要在tensorflow的安装路径下,运行python

>>> import tensorflowas tf

>>> hello = tf.constant('Hello, TensorFlow!')

>>> sess = tf.Session()

>>> print(sess.run(hello))

输出Hello, TensorFlow!


如何阅读tensorflow源码

https://blog.csdn.net/langb2014/article/details/69258090


作者: huang_yx005
原文链接:https://blog.csdn.net/huang_yx005/article/details/78873134

  • 发表于 2019-11-26 11:43
  • 阅读 ( 14 )
  • 分类:tensorflow

0 条评论

请先 登录 后评论
不写代码的码农
huang_yx005 _csdn

0 篇文章

作家榜 »

  1. AI君 10 文章
  2. Tzung-Wen Liau 0 文章
  3. blairan 0 文章
  4. rookie 0 文章
  5. 陈凯 0 文章
  6. huanxue 0 文章
  7. admin 0 文章
  8. Lzs1998_csdn 0 文章