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. Что я могу сделать?

0

Решение

Не могли бы вы попытаться отключить CuDNN 7 в вашей среде и попытаться поместить файл в папку cudnn6 в соответствующую папку в папке CUDA.

Это работает для меня, надеюсь, это поможет.

0

Другие решения

Других решений пока нет …

По вопросам рекламы ammmcru@yandex.ru
Adblock
detector