C++ версия TensorFlow
Мне нужно использовать C++ API TensorFlow. Поэтому кажется необходимым собрать TensorFlow из источников, поэтому я делаю это в соответствии с https://www.tensorflow.org/install/install_sources. Я хочу иметь версию с поддержкой GPU. Я запускаю./configure с параметрами по умолчанию.
Однако при запуске
bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package
возникает следующая ошибка:
Ошибка конфигурации Cuda: версия cuDNN, обнаруженная в /usr/lib/x86_64-linux-gnu/include/cudnn.h (7.0.3), не соответствует TF_CUDNN_VERSION (6)
Ранее я установил CuDNN 7, но затем установил cudnn 6 для использования с tenorflow. Что я могу сделать?
1 ответ
Не могли бы вы попытаться отключить CuDNN 7 в вашей среде и попытаться поместить файл в папку cudnn6 в соответствующую папку в папке CUDA.
Это работает для меня, надеюсь, это поможет.