Поддерживается ли tenorflow-gpu (= <1,14) с CUDA и cuDDN в Red Hat Enterprise Linux 7/8 или цент 7? - PullRequest
0 голосов
/ 26 сентября 2019

У меня есть cent os 7 с Nvidia 2080 super graphic, и я установил следующее:

  1. Графический драйвер 430.4
  2. CUDA 10.1
  3. CuDNN 7.6
  4. conda env python 3.7
  5. tenorflow-gpu 1.14 (pip install tenorflow-gpu)

, затем pip прививает необходимые библиотеки для кода.

но когда я попытался запустить этот код ссылка я получаю ошибку инициализации cuDNN ..

Любое руководство или предложения очень ценятся :-)

enter image description here

Ответы [ 2 ]

0 голосов
/ 27 сентября 2019

Если вы используете среду conda, вам следует установить пакеты с conda.Попробуйте установить tenorflow-gpu с помощью conda;он должен позаботиться обо всех ваших зависимостях, кроме драйвера NVidia, который у вас уже есть.Подобные действия позволяют conda позаботиться о дополнительных зависимостях пакетов, таких как инструментарий CUDA и cudnn, помимо прочего.

Источник: Устанавливая только версию 10.1 набора инструментов CUDA (с драйвером) в системное местоположение и тензор-gpu в среде anaconda, у меня есть тензор-поток-gpu 1.14, работающий с Cuda 10.1 при установке CentOS 7а также на установке RHEL7.

0 голосов
/ 26 сентября 2019

Я не могу сказать о CentOS, так как я никогда не устанавливал на него, но после просмотра информации, которую вы предоставили.Ошибка в вашей картинке говорит, что Cudnn не может быть инициализирован, это скорее всего проблемы совместимости.Вы должны соблюдать эти требования.Итак, полностью удалите Cuda 10.1 и установите Cuda 10. Проверьте, сохраняется ли проблема.

...