Отсутствующие библиотеки при попытке использовать графический процессор с Tensorflow
Я пытаюсь использовать графический процессор с 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] Could not load dynamic library 'libnvinfer.so.6'; dlerror: libnvinfer.so.6: cannot open shared object file: No such file or directory; 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] Could not load dynamic library 'libnvinfer_plugin.so.6'; dlerror: libnvinfer_plugin.so.6: cannot open shared object file: No such file or directory; 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] Cannot dlopen some TensorRT libraries. If you would like to use Nvidia GPU with TensorRT, please make sure the missing libraries mentioned above are installed properly.
Подскажите, пожалуйста, как мне получить недостающие библиотеки?
Спасибо Хадил
1 ответ
Это работает в Ubuntu 19.10 (и, вероятно, в других версиях):
ВНИМАНИЕ: Это может привести к перегрузке вашего графического драйвера NVIDIA, поэтому вы можете вернуться к загрузке с резервного диска, моментального снимка ZFS или чего-то еще. Вы были предупреждены.
Я смог загрузить графическую версию, но 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
# Install TensorRT. Requires that libcudnn7 is installed above.
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
Затем перезагрузите компьютер и попробуйте импортировать тензорный поток и запустить приложение OpenGL. Оба должны работать, если все пойдет хорошо.