Bulding Tensorflow with bazel:包括TensorRT集管

我正在EC2实例上编译tensorflow,以启用本机指令集。我对bazel完全陌生,对如何配置构建感到不知所措。

我唯一的问题是编译无法使用我放在/usr/local/TensorRT-8.6.1.6的本地安装的TensorRT中的头文件

首先,我尝试在PATH中添加TensorRT-8.6.1.6/include,没有成功。仍然找不到标题。

然后我试着添加bazel copt -I/usr/lib/TensorRT-8.6.1.6/include ,它抱怨引用了执行根目录之外的路径。

最后,我尝试将该目录符号链接到源代码内部,并使用 -ITensorRT-8.6.1.6/include 最近的错误是

this rule is missing dependency declarations for the following files included by 'tensorflow/compiler/tf2tensorrt/stub/nvinfer_stub.cc':
  'TensorRT-8.6.1.6/include/NvInferRuntimeBase.h'
  'TensorRT-8.6.1.6/include/NvInferVersion.h'
  'TensorRT-8.6.1.6/include/NvInferRuntimePlugin.h'

我不知道为什么像TensorRT这样一个常见的tensorflow依赖项如此难以链接。我在巴泽尔没有足够的知识来包括那些标题。