tensorrt 安装 windows python

windows 必须按照 8 及以上版本

1. 下载

官网：https://developer.nvidia.com/nvidia-tensorrt-download

EA 版本代表抢先体验，GA 是稳定版。

下载同cuda对应的版本

2. 移动包或环境变量

(1)将下载的TensorRT-8.2.0.6文件夹中的include文件移动到你的CUDA文件夹下的include，我这是（C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0），
(2)TensorRT-8.2.0.6文件夹中的lib中的dll与lib文件分别放到CUDA下的bin与lib\x64文件夹中
***********************上面必须注意读第2条，有一个复制是互换
否则添加环境变量

3. pip安装包内的whl文件

安装对应的版本

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

4. 测试代码

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import tensorrt as trt
if __name__ == "__main__":
    print(trt.__version__)
    print("hello trt!!")

运行tensorrt显示找不到nvinfer.dll和nvparsers.dll，要么没有复制，要么环境变量没有添加