TensorRT下载安装

TensorRT下载安装

一 下载地址 https://developer.nvidia.com/nvidia-tensorrt-8x-download

版本8.2.3GA(成熟稳定版)和8.4.0EA（新功能测试版）以后，有C++和python的API，完全等价可以混用

---

二 安装tensorrt 的python版本

pip install tensorrt-8.5.1.7-cp38-none-win_amd64.whl

三 安装pycuda

pip install pycuda -i https://pypi.tuna.tsinghua.edu.cn/simple

pycuda版本需要和cuda版本对应

---

安装onnxsim>=0.4.33, onnxruntime-gpu

pip install  onnxsim onnxruntime-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple

---

TensorRT官方教程：

https://docs.nvidia.com/deeplearning/tensorrt/developer-guide/index.html#overview

https://docs.nvidia.com/deeplearning/tensorrt/api/c_api

https://docs.nvidia.com/deeplearning/tensorrt/api/python_api/

https://github.com/NVIDIA/trt-samples-for-hackathon-cn/tree/master/cookbook