[Ubuntu 세팅] 2. TensorRT 설치

해당세팅은 Ubuntu 초기세팅을 기록하기 위한 용도입니다.

TensorRT를 설치하기 전

* 1) Nvidia-Driver   2) Cuda   3) CuDnNN 등이 세팅 되어있어야한다.

1~3 세팅은 이전 포스트 참조)

[Darknet]Darknet위한 초기세팅(Ubuntu20.04)

 

 

* 포스트 시작전 사용자 버전

Nvidia-Driver: 470

CUDA : 11.4

CuDNN: 8.2.2

 

1) TensorRT 다운 (본인은 8.2버전을 다운)

다운로드: https://developer.nvidia.com/tensorrt-download

CUDA11.4 호환되는 TAR파일을 다운 받기(첫번째 다운)

 

cd Downloads #이동
tar -xvzf TensorRT-7.2.3.4.Ubuntu-18.04.x86_64-gnu.cuda-11.0.cudnn8.1.tar.gz

code ~/.bashrc # 작성자는 visual studio code를 사용

#추가 해주기
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/(사용자 경로)/TensorRT-8.2.0.6/lib

# 소싱
source ~/.bashrc

 

 

 설치

*자신의 파이썬 버전에 맞게 설치 (작성자는 3.8버전)

#python
python3 -m pip install tensorrt-8.2.0.6-cp38-none-linux_x86_64.whl

#uff
cd ../uff 
python3 -m pip install uff-0.6.9-py2.py3-none-any.whl

#graphsurgeon
cd ../graphsurgeon
python3 -m pip install graphsurgeon-0.4.5-py2.py3-none-any.whl

#onnx_graphsurgeon
cd ../onnx_graphsurgeon
python3 -m pip install onnx_graphsurgeon-0.3.12-py2.py3-none-any.whl

 

 

설치 확인

python3
import tensorrt
tensorrt.__version__

 

* Nvidia-Driver   2) Cuda   3) CuDnNN  해당 버전들 호환성만 맞춘다면 문제 없이 설치 가능