使用 Bazel 从源代码构建 TensorFlow 时出错

Question

我正在关注This tutorial 从源代码构建 tensorflow，因为我的 GPU 只有 3.0 CUDA 兼容性（如果您知道在哪里可以找到适合我的构建，请告诉我！）。我使用的是 Ubuntu 17.10 和 Bazel 0.13.0。 CUDA 9.0，CuDnn 1.7.3，nvidia-driver 396版。我正在运行bazel build --config=opt --config=cuda //tensorflow/tools --cxxopt="-D_GLIBCXX_USE_CXX11_ABI=0"

它构建了大部分方式，然后给我一个巨大的错误，说它没有找到各种 CUDA 库。

Answer 1

我最近遇到了类似的问题。最近的 Bazel 0.13 版似乎无法找到 LD_LIBRARY_PATH 中设置的 CUDA 库。我找到了两种解决方法：

1. 将 cuda 库路径添加到 /etc/ld.conf.so（参见https://github.com/tensorflow/tensorflow/issues/15889 末尾的注释），或者
2. 使用 Bazel 版本 0.12 而不是 0.13

如果您的系统中只有一个 cuda 安装，第一个有效。但是我为不同的项目使用不同的 cuda 版本，所以我不得不找到另一个解决方案。后一种解决方案似乎从 LD_LIBRARY_PATH 中找到库。