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
'AI | ML > AI 개발 | CUDA' 카테고리의 다른 글
[Onnx] C++ GPU Inference (2) | 2021.12.02 |
---|---|
[TensorRT] Windows 환경 onnx -> trt 변경 (0) | 2021.12.01 |
[TensorRT] EA vs GA vs RC 차이 (1) | 2021.12.01 |
[C++] onnx 결과값 Mat 타입으로 변경 (0) | 2021.11.30 |
[Pytorch Lightning] Pytorch-Lightning 사용해보기 (0) | 2021.07.21 |