AI | ML/AI 개발 | CUDA

[TensorRT] Windows 10 C++ 설치

깜태 2021. 12. 1. 14:52
728x90

나는 CUDA, CUDNN 이 설치되어 있다는 가정하에서 진행하였다.  ( CUDA : 11.0, CUDNN: 8.02, VS : 2019 )

CUDA 환경이나, tensorflow, pytorch 등의 환경이 다르다면 아래의 링크를 기준으로 다른 버전을 참고해야 한다.

https://docs.nvidia.com/deeplearning/tensorrt/release-notes/index.html

 

1. tensorRT 홈페이지에 들어가 환경에 맞는 zip 파일을 다운로드 받는다. (나는 8.2.1 GA 버전을 설치하였다) 

https://developer.nvidia.com/nvidia-tensorrt-download

 

2. 적당한 곳에 압축을 푼다. 

3. Visual Studio를 열고, 속성 페이지에 들어간다.

 

 

4. C/C++ -> 일반 -> 추가 포함 디렉토리에 다운받은 경로/include 를 넣는다

ex) D:\TensorRT-8.2.1.8\include

 

5. 링커 -> 일반 -> 추가 라이브러리 디렉토리 에 경로/lib 을 입력

 

 

6. 해당 프로젝트 파일에 경로/bin 에 있는 .dll 파일들을 추가 후, 샘플 코드 작성 후 테스트

 

 

참고

https://docs.nvidia.com/deeplearning/tensorrt/install-guide/index.html#downloading

728x90