在Windows10下安装TensorRT +onnx

参考:  https://www.jianshu.com/p/538b467f1105

 

1.查看显卡支持的cuda:

 

2.下载TensorRT-5.1.5.0.Windows10.x86_64.cuda-10.0.cudnn7.5

3.下载并安装cuda_10.0.130_411.31_win10.exe

4.下载cudnn-10.0-windows10-x64-v7.5.0.56

5.把cudnn64_7.dll拷贝到cuda9.0的安装目录下bin文件夹中，本文的路径是：C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin

6. 将TensorRT解压位置\lib下的dll文件复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin目录下

 7.添加“环境变量”到path:将TensorRT解压位置\lib 加入系统环境变量

 

安装onnx

1.pip install onnx