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

Question

我正在关注This tutorial从源代码构建张量流，因为我的GPU只有3.0 CUDA兼容性（如果你知道在哪里可以找到适用于我的构建，请告诉我！）。我使用的是Ubuntu 17.10和Bazel 0.13.0。 CUDA 9.0，CuDnn 1.7.3，nvidia-driver version 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中找到了库。