tensorflow离线源码安装

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

1.tensorflow的外部依赖

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

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

bazel fetch //...

为能够离线安装tensorflow,需要建立内部局域网的http服务。(建立http服务可以使用wampserver,可参考 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

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值