Win10安装Tensorrt和torch2trt
1 前言
1.1 实验环境
(1)主机系统:Win10
(2)显卡: Quadro RTX 6000
(3)CUDA版本:V11.7.64
(4)Python:3.8.18
(5)TensorRT:8.6.1.6
1.2 下载链接
(1)TensorRT8.6.1.6下载链接(下载资源包): https://developer.nvidia.com/tensorrt
(2)pycuda下载链接((pip 离线安装):https://pypi.org/project/pycuda/2022.1/#history
也可在线下载:pip install pycuda;用于从 Python 访问Nvidia的CUDA并行计算 API。
(3)torch2trt下载链接((pip 离线安装):https://github.com/NVIDIA-AI-IOT/torch2trt
(4)在线安装onnx和cython: pip install onnx cython
2 具体安装
2.1 TensorRT安装
2.1.1 设置环境变量
1、 解压下载的资源包,并将TensorRT对应的li