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. Что я могу сделать?
Решение
Не могли бы вы попытаться отключить CuDNN 7 в вашей среде и попытаться поместить файл в папку cudnn6 в соответствующую папку в папке CUDA.
Это работает для меня, надеюсь, это поможет.
Другие решения
Других решений пока нет …