-
Windows 10 PC에 TensorRT 설치기AI/Etc 2021. 1. 1. 21:58
1. 사용환경
- PC Spec
CPU AMD Ryzen 5 2600x RAM Essencore Klevv DDR4-17000 8GB x2
SAMSUNG DDR4-21300 16GB x1VGA ZOTAC GAMING Geforce GTX 1660 Ti AMP D6 6GB - CUDA Version : v10.2.89
- cuDNN Version : v8.0.5.39
2. 설치 방법
- 준비물
TensorRT Zip File : https://developer.nvidia.com/nvidia-tensorrt-download
CUDA 10.2 버전의 TensorRT로 진행하겠다. pip install C:\TensorRT-7.2.2.3\graphsurgeon\graphsurgeon-0.4.5-py2.py3-none-any.whl
pip install C:\TensorRT-7.2.2.3\uff\uff-0.6.9-py2.py3-none-any.whl
pip install C:\TensorRT-7.2.2.3\onnx_graphsurgeon\onnx_graphsurgeon-0.2.6-py2.py3-none-any.whl