在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