win10에 TensorRT 설치

TensorRT 다운로드

자신의 CUDA 버전에 따라 해당 TensorRT 압축 패키지를 다운로드하십시오. 저는 NVIDIA GeForce 3090, CUDA11.1을 사용하고 nvcc -VCUDA 버전을 확인하는 데 사용합니다.

(base) C:\Users\Administrator>nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2020 NVIDIA Corporation
Built on Mon_Oct_12_20:54:10_Pacific_Daylight_Time_2020
Cuda compilation tools, release 11.1, V11.1.105
Build cuda_11.1.relgpu_drvr455TC455_06.29190527_0

아래 그림과 같이 Nvidia 공식 웹사이트 에 들어가 해당 TensorRT 버전을 다운로드합니다.

압축을 풀다

Visual Studio를 사용하여 TensorRT에서 samples\sampleMNIST 프로젝트를 엽니다. 나는 VS2017을 사용하고, 연 후 다음과 같이 구성하고 마우스 오른쪽 버튼 클릭 → 속성 TensorRT의 lib 폴더에는 dll 파일과 lib 파일이 포함되어 있으므로 TensorRT의 lib 폴더 경로에 다음 구성을 추가합니다. VC++ 디렉터리 → 실행 파일 디렉터리 다음과 같이 TensorRT에 lib 폴더 경로를 추가합니다: VC++ 디렉토리 → 라이브러리 디렉토리 다음과 같이 TensorRT에 포함 폴더 경로를 추가합니다: C/C++→일반→추가 포함 디렉터리 다음과 같이 lib 폴더에 모든 lib 파일을 추가합니다[nvinfer.lib, nvinfer_plugin.lib, nvonnxparser.lib, nvparsers.lib]: 링커 → 입력 → 추가 종속성 프로젝트를 마우스 오른쪽 버튼으로 클릭 → 재생성

TensorRT에 bin 폴더를 입력하고 cmd에서 sample_mnist.exe를 실행하면 아래와 같이 뜨면 설치가 성공한 것입니다.