Tensorflow中8位量化的致命错误

Question

我正在尝试使用Bazel和我的Ubuntu 16.04系统在Tensorflow中运行量化模型。

我运行了以下命令：

bazel build tensorflow/tools/quantization:quantize_graph

这是错误：

  

错误：/ tensorflow-master / tensorflow / contrib / verbs / BUILD：136：1：规则'// tensorflow / contrib / verbs：rdma'的C ++编译失败（退出1）。

     

在tensorflow / contrib / verbs / rdma.cc中包含的文件中：18：0：   ./tensorflow/contrib/verbs/rdma.h:21:30：致命错误：infiniband / verbs.h：没有这样的文件或目录

     

编译终止。

     

目标// tensorflow / tools / quantization：quantize_graph无法构建

我该如何解决这个问题？提前谢谢！

Answer 1

尝试使用

安装libibverbs-dev

sudo apt-get update
sudo apt-get install libibverbs-dev

Answer 2

使用

安装libibverbs-dev和librdmacm-dev

sudo apt-get install libibverbs-dev
sudo apt-get install librdmacm-dev