如何强制 tensorflow 使用自定义版本的 Eigen？

Question

我正在编译 Tensorflow 1.5，我想强制 bazel 包含一个自定义版本的 eigen 头文件，这些文件位于： usr/local/lib/python2.7/dist-packages/... 相反，每当我尝试编译时（即使在 bazel clean --expunge 之后），tensorflow 都会使用不同的文件，这些文件在构建过程中被复制到： /root/.cache/bazel/_bazel_root/

有没有办法强制tensorflow使用不同的文件？

Answer 1

您可以将tensorflow/workspace.bzl 中eigen_archive 的tf_http_archive 规则（不得更改名称）更改为new_local_repository，并使用Tensorflow 的特征BUILD 文件（//third_party:eigen.BUILD）。