Ошибка построения TensorFlow из источника с помощью Bazel - PullRequest
Новая
       29

Ошибка построения TensorFlow из источника с помощью Bazel

0 голосов
/ 01 мая 2018

Я следую Это руководство по созданию тензорного потока из исходного кода, потому что мой GPU поддерживает только 3.0 CUDA (Если вы знаете, где найти сборку, которая будет работать для меня, дайте мне знать!). Я использую Ubuntu 17.10 и Bazel 0.13.0. CUDA 9.0, CuDnn 1.7.3, версия драйвера nvidia 396. Я работаю bazel build --config=opt --config=cuda //tensorflow/tools --cxxopt="-D_GLIBCXX_USE_CXX11_ABI=0"

Он строит большую часть пути, а затем дает мне огромную ошибку, говоря, что не нашел различных библиотек CUDA.

1 Ответ

0 голосов
/ 14 мая 2018

Недавно я столкнулся с подобными проблемами. Похоже, что последняя версия Bazel 0.13 не может найти библиотеки CUDA, установленные в LD_LIBRARY_PATH. Я нашел два обходных пути:

  1. Добавить пути к библиотекам cuda в /etc/ld.conf.so (см. Комментарий в конце https://github.com/tensorflow/tensorflow/issues/15889), или
  2. Используйте версию Bazel 0.12 вместо 0.13

Первый из них работает, если в вашей системе установлена ​​только одна установка cuda. Однако я использую разные версии cuda для разных проектов, поэтому мне пришлось искать другое решение. Последнее решение, похоже, находит библиотеки из LD_LIBRARY_PATH.

...