본문 바로가기

Tect.tip/Linux

[install] TensorRT 설치

TensorRT download

https://developer.nvidia.com/tensorrt

tar -xzvf TensorRT-7.2.1.6.Ubuntu-18.04.x86_64-gnu.cuda-10.2.cudnn8.0.tar.gz

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda-10.2/lib64:/usr/local/cuda/extras/CUPTI/lib64:/home/name/TensorRT
7.2.1.6/lib"

whl file install

 cd TensorRT-5.1.x.x/python # for python 2.7 
 sudo pip2 install tensorrt-5.1.x.x-cp27-none-linux_x86_64.whl # for python 3.x 
 sudo pip3 install tensorrt-5.1.x.x-cp3x-none-linux_x86_64.whl 
 cd TensorRT-5.1.x.x/uff # for python 2.7 
 sudo pip2 install uff-0.6.3-py2.py3-none-any.whl # for python 3.x 
 sudo pip3 install uff-0.6.3-py2.py3-none-any.whl 
 which convert-to-uff 
 cd TensorRT-5.1.x.x/graphsurgeon # for python 2.7 
 sudo pip2 install graphsurgeon-0.4.0-py2.py3-none-any.whl # for python 3.x 
 sudo pip3 install graphsurgeon-0.4.0-py2.py3-none-any.whl