python – 安装Tensorflow并提供量化支持

这是我对另一个问题的跟进:Error with 8-bit Quantization in Tensorflow

基本上,我想安装具有8位量化支持的Tensorflow.目前,我在CentOS 7机器上安装了Tensorflow 0.9和pip安装方法(没有GPU支持).

我可以编译并运行Pete Warden博客文章中给出的代码.但是,我无法导入Pete Warden回复中给出的功能.我想添加量化支持.我在Tensorflow文档中也找不到有关量化部分的任何细节.

有人可以分享一下如何做的细节吗?

解决方法:

暂时,我可以找到一种方法来做到这一点.但仍在等待任何TensorFlow开发人员的官方方法.

>首先安装tensorflow(我尝试了源安装以及PIP安装,两者都很好)
>从Github仓库获取tensorflow源并转到tensorflow根目录(我将其称为tensorflow_root.
>现在编译Pete Warden’s blog中给出的量化脚本

bazel build tensorflow / contrib /量化/工具:quantize_graph

这将为量化版本创建ops库.转到tensorflow_root / bazel-bin / tensorflow / contrib / quantization,您应该看到两个库文件:_quantized_ops.so和kernels / _quantized_kernels.so

>现在,在脚本中,连同tensorflow,您还应使用专用的tensorflow函数导入这两个库文件

你可以使用tf.load_op_library()函数来完成它

import tensorflow as tf
qops = tf.load_op_library('[tensorflow_root]/bazel-bin/tensorflow/contrib/quantization/_quantized_ops.so')
qkernelops = tf.load_op_library('[tensorflow_root]/bazel-bin/tensorflow/contrib/quantization/kernels/_quantized_kernels.so')
上一篇:套索回归


下一篇:小白专场-树的同构-python语言实现