首页 文章

Tensorflow中8位量化的致命错误

提问于
浏览 次
5

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

我运行了以下命令：

bazel build tensorflow/tools/quantization:quantize_graph

这是错误：

错误：/ tensorflow-master / tensorflow / contrib / verbs / BUILD：136：1：C编译规则'/ tensorflow / contrib / verbs：rdma'失败（退出1） . 在tensorflow / contrib / verbs / rdma.cc中包含的文件中：18：0：./tensorflow/contrib/verbs/rdma.h：21：30：致命错误：infiniband / verbs.h：没有这样的文件或目录编译终止 . 目标// tensorflow / tools / quantization：quantize_graph无法构建

我怎么解决这个问题？提前致谢！

tensorflow bazel quantization

2 回答

  • 1

    尝试使用安装 libibverbs-dev

    sudo apt-get update
sudo apt-get install libibverbs-dev
    回复于
  • 1

    使用安装 libibverbs-dev 和 librdmacm-dev

    sudo apt-get install libibverbs-dev
sudo apt-get install librdmacm-dev
    回复于

相关问题