在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 FilesNVIDIA GPU Computing ToolkitCUDAv10.0in

6. 将TensorRT解压位置lib下的dll文件复制到C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.0in目录下

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

安装onnx

1.pip install onnx

原文地址:https://www.cnblogs.com/shellycn/p/13049790.html