Installation TensorRT
TensorRT installed here only to show the tar package. Installation Environment:
-
CUDA 10.0, 7.5 cuDNN
-
ubuntu 18.0.4
-
Download TensorRT-5.1.2.2.Ubuntu-14.04.5.x86_64-gnu.cuda-10.0.cudnn7.5.tar.gz
-
Unzip to a specific directory
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
-
Installation TensorRT and related documents
pip install /usr/local/TensorRT-5.1.2.2/python/tensorrt-5.1.2.2-cp36-none-linux_x86_64.whl
pip install /usr/local/TensorRT-5.1.2.2/uff/uff-0.6.3-py2.py3-none-any.whl
pip install /usr/local/TensorRT-5.1.2.2/graphsurgeon/graphsurgeon-0.4.0-py2.py3-none-any.whl
Will/usr/local/TensorRT-5.1.2.2/targets/x86_64-linux-gnu/lib
add to/etc/ld.so.conf
, and thensudo ldconfig
If you run
In [1]: import tensorrt as trt
In [2]:
No problems with the installation was successful. If you are prompted libnvinfer.so
not find most of the third step is not performed.