Отсутствующие библиотеки при попытке использовать GPU с Tensorflow

Я пытаюсь использовать GPU с Tensorflow, поэтому я установил CUDA 10.1, CUNN 7.6, TensorRT 6 и Tensorflow-GPU 2.1.0. Но когда я импортирую tensorflow, чтобы проверить, может ли он обнаружить gpu, возникает следующая ошибка

>>> import tensorflow as tf
2020-04-22 14:44:14.473830: W tensorflow/stream_executor/platform/default/dso_loader.cc:55] Не удалось загрузить динамическую библиотеку 'libnvinfer.so.6'; dlerror: libnvinfer.so.6: не удается открыть общий файл объекта: Нет такого файла или директории; LD_LIBRARY_PATH: /usr/local/cuda/lib64:/usr/local/cuda/lib64:/usr/local/cuda/lib64:/usr/local/cuda-10.1/lib64
2020-04-22 14:44:14.473880: W tensorflow/stream_executor/platform/default/dso_loader.cc:55] Не удалось загрузить динамическую библиотеку 'libnvinfer_plugin.so.6'; dlerror: libnvinfer_plugin.so.6: не удается открыть общий файл объекта: Нет такого файла или директории; LD_LIBRARY_PATH: /usr/local/cuda/lib64:/usr/local/cuda/lib64:/usr/local/cuda/lib64:/usr/local/cuda-10.1/lib64
2020-04-22 14:44:14.473887: W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:30] Не удалось загрузить некоторые библиотеки TensorRT. Если вы хотите использовать Nvidia GPU с TensorRT, убедитесь, что вышеупомянутые недостающие библиотеки установлены правильно.

Можете сказать, как я могу получить недостающие библиотеки, пожалуйста?

Спасибо
Хадиль

Это работает в Ubuntu 19.10 (и вероятно в других версиях):

ПРЕДУПРЕЖДЕНИЕ: Это может повредить ваш графический драйвер NVIDIA, поэтому лучше сделайте резервную копию системы на случай, если что-то пойдет не так. Вы были предупреждены.

Я смог загрузиться в графической версии, но OpenGL не работал. Инструкции о том, как я это восстановил, приведены ниже.

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1810/x86_64/cuda-repo-ubuntu1810_10.1.168-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1810_10.1.168-1_amd64.deb
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1810/x86_64/7fa2af80.pub
sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/7fa2af80.pub
sudo apt update
wget http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/nvidia-machine-learning-repo-ubuntu1804_1.0.0-1_amd64.deb
sudo dpkg -i nvidia-machine-learning-repo-ubuntu1804_1.0.0-1_amd64.deb
sudo apt update

sudo apt-get install --no-install-recommends \
    cuda-10-1 \
    libcudnn7=7.6.4.38-1+cuda10.1  \
    libcudnn7-dev=7.6.4.38-1+cuda10.1

    # Установите TensorRT. Требуется, чтобы вышеупомянутый libcudnn7 был установлен.
sudo apt-get install -y --no-install-recommends  
    libnvinfer6=6.0.1-1+cuda10.1 \
    libnvinfer-dev=6.0.1-1+cuda10.1 \
    libnvinfer-plugin6=6.0.1-1+cuda10.1

Как я уже сказал выше, мне пришлось переустановить драйвера для моей видеокарты NVIDIA:

sudo add-apt-repository ppa:graphics-drivers
sudo apt update
sudo apt -u dist-upgrade

Затем перезагрузите и попробуйте как импортировать tensorflow, так и запустить приложение OpenGL. Оба должны работать, если все прошло хорошо.

Я решил проблему следующим образом

1. Сначала нам нужно найти местоположение файлов “libnvinfer.so”

sudo find . -iname libnvinfer.so*

Я получил следующий результат:

./usr/lib/x86_64-linux-gnu/libnvinfer.so
./usr/lib/x86_64-linux-gnu/libnvinfer.so.6
./usr/lib/x86_64-linux-gnu/libnvinfer.so.7.2.1
./usr/lib/x86_64-linux-gnu/libnvinfer.so.7
./usr/lib/x86_64-linux-gnu/libnvinfer.so.6.0.1

Необходимые файлы находятся в /usr/lib/x86-linux-gnu

2. Нам нужно добавить этот путь в $LD_LIBRARY_PATH (переменная окружения)

   echo “export LD_LIBRARY_PATH=/usr/lib/x86_64-linux-gnu:$LD_LIBRARY_PATH” >> ~/.bashrc

   source ~/.bashrc