TensorRTインストール

インストールTensorRT

TensorRTは唯一のtarパッケージを表示するためにここにインストールされています。インストール環境:

  • CUDA 10.0、7.5 cuDNN

  • Ubuntuの18.0.4

  • TensorRT-5.1.2.2.Ubuntu-14.04.5.x86_64-gnu.cuda-10.0.cudnn7.5.tar.gzをダウンロード

  • 特定のディレクトリに解凍しますsudo tar xzvf TensorRT-5.1.2.2.Ubuntu-14.04.5.x86_64-gnu.cuda-10.0.cudnn7.5.tar.gz -C /usr/local

  • インストールTensorRTおよび関連文書

    1. pip install /usr/local/TensorRT-5.1.2.2/python/tensorrt-5.1.2.2-cp36-none-linux_x86_64.whl
    2. pip install /usr/local/TensorRT-5.1.2.2/uff/uff-0.6.3-py2.py3-none-any.whl
    3. pip install /usr/local/TensorRT-5.1.2.2/graphsurgeon/graphsurgeon-0.4.0-py2.py3-none-any.whl
      ウィル/usr/local/TensorRT-5.1.2.2/targets/x86_64-linux-gnu/libに追加し/etc/ld.so.confた後、およびsudo ldconfig

あなたが実行した場合

In [1]: import tensorrt as trt

In [2]: 

インストールに問題は成功しませんでした。プロンプトが表示された場合はlibnvinfer.so第三の工程のほとんどが行われていません見つけることができません。

65元記事公開 ウォン称賛26 ビューに10万+を

おすすめ

転載: blog.csdn.net/bleedingfight/article/details/88896648