参考: 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