エラーCould not load dynamic library libnvinfer.so.6解決方法

TensorFlowの可用性をテストするとき(tf.test.is_gpu_available()にエラーが発生しました.

エラーの原因

TensorRTがインストールされていません

NVIDIA TensorRTについて:

NVIDIA TensorRTは高性能ニューラルネットワーク推理(Inference)エンジンであり、生産環境に深度学習アプリケーションを配置するために使用され、画像分類、分割、目標検出などが適用され、最大の推理スループットと効率を提供することができる.TensorRTは、既存および将来のネットワークアーキテクチャを加速させるプログラム可能な最初の推理加速器です.TensorRTにはCUDAのサポートが必要です.TensorRTには、本番環境に配置された深度学習モデルを最適化するために作成されたライブラリが含まれており、トレーニングされたニューラルネットワーク(通常32ビットまたは16ビットのデータを使用する)を取得し、精度を低下させるINT 8演算に対してこれらのネットワークを最適化することができる.CUDAのプログラマブル性により、TensorRTは、深さニューラルネットワークの多様化、複雑な成長傾向を加速させることができる.TensorRTの大幅な加速により、サービス・プロバイダは、これらのコンピューティング・クリティカルな人工知能ワークロードをコスト・パフォーマンスの高いコストで導入できます.

エラー形式

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/nvidia/lib:/usr/local/nvidia/lib64:/usr/local/cuda/lib64/

実際には、これはエラーではなく警告情報にすぎません.Could not load dynamic library'libnvinfer.so.6'とlibnvinfer.so.6は、libnvinfer_plugin.so.6を使用する場合にのみ必要です.この2つのライブラリがなくてもtensorflowを実行できます.
-引用-https://blog.csdn.net/toopoo/article/details/104506143

ソリューション

TensorRTのインストール

# Install TensorRT.
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