ABOUT ME

-

Today
-
Yesterday
-
Total
-
  • 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 x1 
    VGA 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

    댓글

Designed by Tistory.