При использовании тензор-gpu cudnn не удается в течение времени ссылки - PullRequest
Новая
       32

При использовании тензор-gpu cudnn не удается в течение времени ссылки

0 голосов
/ 03 марта 2019

Я пытаюсь использовать tensorflow-gpu в Ubuntu.После серии установки происходит сбой, когда я импортирую tenorflow в python3Ошибка: 

ImportError: /$MYPATH$/python3.6/site-packages/tensorflow/python/../libtensorflow_framework.so: 
symbol cudnnConvolutionBiasActivationForward version libcudnn.so.7 not defined in file libcudnn.so.7 with link time reference

Моя среда выглядит следующим образом:

  • cuda: 9.0.176
  • cudnn: 7,2
  • tensorflow-gpu: 1.10

Кроме того, я добавил путь к cuda в ~/.bashrc как: 

export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH

export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

Я думаю, что проблема связана с неправильной ссылкой на файл cudnn, но яНе знаю, как найти точную проблему и исправить ее.

Как я могу решить эту проблему?Спасибо за вашу помощь!

1 Ответ

0 голосов
/ 03 марта 2019

Причина, по которой cudnn не может найти нужную ссылку, заключается в том, что я использую anaconda среду.Вместо того, чтобы устанавливать много пакетов вручную, я просто установил tensorflow-gpu, используя: 

$conda create --name tf_gpu tensorflow-gpu
$source activate tf_gpu

Это работает!Учебник находится здесь .

Но причина, по которой я могу не использовать pip install tensorflow-gpu для правильной установки, до сих пор неизвестна. Все же нужна ваша помощь и объяснение!

...