Không thể tải thư viện động 'libnvinfer.so.6'

Tôi đang cố gắng nhập gói python TensorFlow bình thường, nhưng tôi gặp lỗi sau:

Đây là văn bản từ hình ảnh thiết bị đầu cuối ở trên:

2020-02-23 19:01:06.163940: 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
2020-02-23 19:01:06.164019: 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
2020-02-23 19:01:06.164030: 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.
<module 'tensorflow_core._api.v2.version' from '/home/saman/miniconda3/envs/testconda/lib/python3.7/site-packages/tensorflow_core/_api/v2/version/__init__.py'

Đây là một cảnh báo, không phải là một lỗi. Bạn vẫn có thể sử dụng TensorFlow. Các thư viện dùng chung libnvinfervà libnvinfer_pluginchỉ là tùy chọn và chỉ bắt buộc nếu bạn đang sử dụng các khả năng TensorRT của nvidia.

Hướng dẫn cài đặt của TensorFlow liệt kê các phụ thuộc GPU:

Phần mềm NVIDIA® sau phải được cài đặt trên hệ thống của bạn:

• Trình điều khiển GPU NVIDIA® CACDADA 10.1 yêu cầu 418.x trở lên.
• Bộ công cụ CUDA® Hướng dẫnTensorFlow hỗ trợ CUDA 10.1 (TensorFlow> = 2.1.0)
• CUPTI xuất xưởng với Bộ công cụ CUDA.
• SDK cuDNN (> = 7.6)
• (Tùy chọn) TensorRT 6.0 để cải thiện độ trễ và thông lượng cho suy luận trên một số mô hình.

Bạn có thể cài đặt chúng trên Ubuntu 18.04 bằng các lệnh sau (được lấy từ tài liệu TensorFlow ):

# Add NVIDIA package repositories
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-repo-ubuntu1804_10.1.243-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1804_10.1.243-1_amd64.deb
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
sudo apt-get 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 apt install ./nvidia-machine-learning-repo-ubuntu1804_1.0.0-1_amd64.deb
sudo apt-get update

# Install NVIDIA driver
sudo apt-get install --no-install-recommends nvidia-driver-430
# Reboot. Check that GPUs are visible using the command: nvidia-smi

# Install development and runtime libraries (~4GB)
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

Hầu hết các tin nhắn này là cảnh báo, không phải lỗi. Chúng chỉ có nghĩa là các thư viện sử dụng GPU Nvidia chưa được cài đặt, nhưng bạn không cần phải có GPU Nvidia để sử dụng Tensorflow nên bạn không cần các thư viện này. Nhận xét của jakub cho biết cách tắt các cảnh báo:

export TF_CPP_MIN_LOG_LEVEL="2"

Tuy nhiên, tôi cũng chạy Tensorflow mà không có công cụ Nvidia và có thêm một thông báo là lỗi, không phải cảnh báo:

2020-04-10 10:04:13.365696: E tensorflow/stream_executor/cuda/cuda_driver.cc:351] failed call to cuInit: UNKNOWN ERROR (303)

Nó không liên quan vì nó cũng đề cập đến cuda, vốn dành cho Nvidia. Nó dường như không phải là một lỗi nghiêm trọng.

Tôi nhận được cảnh báo này là kết quả của việc cập nhật (tình cờ) gói libvnifer6. Nó đã được cập nhật 6.0.1-1+cuda10.2trong khi cài đặt ban đầu được sử dụng 6.0.1-1+cuda10.1.

Sau khi tôi gỡ cài đặt các gói tham chiếu cuda10.2và chạy lại

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

cảnh báo này đã biến mất.