-2

我正在按照本教程从源代码构建 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 库。

4

1 回答 1

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 中找到库。

于 2018-05-14T09:43:15.643 回答